Skip to content

Commit 17f7d72

Browse files
committed
fix ut
1 parent 8f38df5 commit 17f7d72

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tf2onnx/onnx_opset/tensor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1767,8 +1767,8 @@ class MatrixDiagPart:
17671767
@classmethod
17681768
def version_11(cls, ctx, node, **kwargs):
17691769
# MatrixDiagPart by slice and gather
1770-
minus_two_one, minus_two, minus_one, zeo, zeo_zeo, one, two = \
1771-
ctx.make_consts([[-2, -1], [-2], [-1], [0], [0, 0], [1], [2]])
1770+
minus_two_one, minus_two, minus_one, zeo, zeo_zeo, one, two, two_one = \
1771+
ctx.make_consts([[-2, -1], [-2], [-1], [0], [0, 0], [1], [2], [2, 1]])
17721772
zeo_, one_ = ctx.make_consts([0, 1])
17731773

17741774
input_shape = ctx.make_node('Shape', [node.input[0]])

0 commit comments

Comments
 (0)