Skip to content

Commit 13902dc

Browse files
Merge pull request #1139 from onnx/tom/FixLargeModelBug
Fixed bug with order of args when make large models
2 parents 737cb68 + 24afefa commit 13902dc

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
@@ -384,7 +384,8 @@ def update_proto(self, external_tensor_storage=None):
384384
attr_graphs = self.get_body_graphs()
385385
if attr_graphs:
386386
for attr_name, sub_graph in attr_graphs.items():
387-
graph_proto = sub_graph.make_graph("graph for " + self.name + " " + attr_name, external_tensor_storage)
387+
graph_proto = sub_graph.make_graph("graph for " + self.name + " " + attr_name,
388+
external_tensor_storage=external_tensor_storage)
388389
self.set_attr(attr_name, graph_proto)
389390

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

0 commit comments

Comments
 (0)