Skip to content

Commit 701dbfe

Browse files
committed
fix math lrn op numerical problem
1 parent 5d58e9c commit 701dbfe

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tf2onnx/onnx_opset/math.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -262,7 +262,7 @@ def version_1(cls, ctx, node, **kwargs):
262262

263263
ctx.insert_new_node_on_input(node, "Transpose", node.input[0], perm=[0, 3, 1, 2])
264264
op_name = utils.make_name(node.name)
265-
ctx.insert_new_node_on_output("Transpose", node.output[0], perm=[0, 2, 3, 1],name=op_name)
265+
ctx.insert_new_node_on_output("Transpose", node.output[0], perm=[0, 2, 3, 1], name=op_name)
266266

267267

268268
@tf_op(["MatMul", "BatchMatMul"])

0 commit comments

Comments
 (0)