Skip to content

Commit f4f2f04

Browse files
authored
Merge pull request #934 from hfinger/fix-transp-opt
fix transpose optimizer for slice op
2 parents 43c01e5 + dfd18c8 commit f4f2f04

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tf2onnx/optimizer/transpose_optimizer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -645,7 +645,7 @@ def _slice_handler(self, trans, node):
645645
if axes == [0, 1, 2, 3]:
646646
# axes node might be shared
647647
new_axes = np.array(NCHW_TO_NHWC, dtype=np.int64)
648-
if self._nodes_has_single_consumer_node([node]):
648+
if self._nodes_has_single_consumer_node([node.inputs[3]]):
649649
node.inputs[3].set_tensor_value(new_axes)
650650
else:
651651
new_axes_const = self._g.make_const(

0 commit comments

Comments
 (0)