We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a918f00 commit a6f3cbfCopy full SHA for a6f3cbf
tf2onnx/optimizer/reshape_optimizer.py
@@ -109,4 +109,5 @@ def get_reshape_dim(val, i, shift):
109
if shift < 0:
110
unsqueeze_node = GraphBuilder(graph).make_unsqueeze({'data': node.input[0], 'axes': list(range(-shift))})
111
graph.replace_inputs(node, [unsqueeze_node, const_shape])
112
+
113
return True
0 commit comments