Skip to content

Commit 2883580

Browse files
Fix bug in einsum optimizer reading attributes (#1652)
Signed-off-by: Tom Wildenhain <[email protected]>
1 parent 04d2488 commit 2883580

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

tf2onnx/optimizer/einsum_optimizer.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,12 +90,20 @@ def _preprocess(a, axis):
9090
return (np.concatenate(targs, axis),)
9191

9292
def _op_gemm(self, a, b, c=None, alpha=None, beta=None, # pylint: disable=C0103
93-
transA=False, transB=False): # pylint: disable=C0103
93+
transA=None, transB=None): # pylint: disable=C0103
9494
"Runtime for operator."
9595
if alpha is not None:
9696
alpha = alpha.f
9797
if beta is not None:
9898
beta = beta.f
99+
if transA is None:
100+
transA = False
101+
else:
102+
transA = transA.i
103+
if transB is None:
104+
transB = False
105+
else:
106+
transB = transB.i
99107

100108
def _gemm00(a, b, c, alpha, beta):
101109
o = np.dot(a, b) * alpha

0 commit comments

Comments
 (0)