Skip to content

Commit 3bf9eae

Browse files
authored
Merge pull request #972 from jignparm/jignparm/deepcopy
Fix graph deepcopy issue
2 parents b732311 + d445c10 commit 3bf9eae

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tf2onnx/graph.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -301,7 +301,7 @@ def set_tensor_value(self, new_val):
301301
self.set_attr("value", onnx_tensor)
302302
# track shapes in _output_shapes
303303
self._graph_check()
304-
self.graph.set_shape(onnx_tensor.name, onnx_tensor.dims)
304+
self.graph.set_shape(onnx_tensor.name, list(onnx_tensor.dims))
305305

306306
def get_body_graphs(self):
307307
self._graph_check()

0 commit comments

Comments
 (0)