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 daeab0d commit 7bafec5Copy full SHA for 7bafec5
tools/onnx-optimize.py
@@ -50,10 +50,7 @@ def main():
50
51
onnx_graph = g.make_graph(org_model_proto.graph.doc_string + " (+tf2onnx/onnx-optimize)")
52
53
- kwargs = {"producer_name": org_model_proto.producer_name,
54
- "producer_version": org_model_proto.producer_version,
55
- "opset_imports": org_model_proto.opset_import,
56
- "ir_version": org_model_proto.ir_version}
+ kwargs = GraphUtil.get_onnx_model_properties(org_model_proto)
57
58
model_proto = helper.make_model(onnx_graph, **kwargs)
59
0 commit comments