Skip to content

Commit a312d15

Browse files
committed
Minor fix
1 parent 01c403b commit a312d15

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tf2onnx/optimizer/transpose_optimizer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -607,8 +607,8 @@ def _pad_handler(self, trans, node):
607607
if input1.is_const():
608608
if input1.data_format in ["NHWC", "unkown"]:
609609
if not self._nodes_has_single_consumer_node([input1]):
610-
input1 = self.g.copy_const(input1)
611-
self.input[1] = input1.output[0]
610+
input1 = self._g.copy_const(input1)
611+
node.input[1] = input1.output[0]
612612
pads = input1.get_tensor_value()
613613
# NHWC->NCHW
614614
new_pads = np.array([pads[0], pads[3], pads[1], pads[2], pads[4], pads[7], pads[5], pads[6]],

0 commit comments

Comments
 (0)