Skip to content

Commit 24afefa

Browse files
Fixed bug with order of args when make large models
Signed-off-by: Tom Wildenhain <[email protected]>
1 parent 38f5a48 commit 24afefa

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tf2onnx/graph.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -382,7 +382,8 @@ def update_proto(self, external_tensor_storage=None):
382382
attr_graphs = self.get_body_graphs()
383383
if attr_graphs:
384384
for attr_name, sub_graph in attr_graphs.items():
385-
graph_proto = sub_graph.make_graph("graph for " + self.name + " " + attr_name, external_tensor_storage)
385+
graph_proto = sub_graph.make_graph("graph for " + self.name + " " + attr_name,
386+
external_tensor_storage=external_tensor_storage)
386387
self.set_attr(attr_name, graph_proto)
387388

388389
attr = list(self.get_onnx_attrs(external_tensor_storage).values())

0 commit comments

Comments
 (0)