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 b732311 commit d445c10Copy full SHA for d445c10
tf2onnx/graph.py
@@ -301,7 +301,7 @@ def set_tensor_value(self, new_val):
301
self.set_attr("value", onnx_tensor)
302
# track shapes in _output_shapes
303
self._graph_check()
304
- self.graph.set_shape(onnx_tensor.name, onnx_tensor.dims)
+ self.graph.set_shape(onnx_tensor.name, list(onnx_tensor.dims))
305
306
def get_body_graphs(self):
307
0 commit comments