Export¶
-
onnx_graphsurgeon.
export_onnx
(graph: onnx_graphsurgeon.ir.graph.Graph, do_type_check=True, **kwargs) → onnx.onnx_ml_pb2.ModelProto¶ Exports an onnx-graphsurgeon Graph to an ONNX model.
- Parameters
graph (Graph) – The graph to export
do_type_check (bool) – Whether to check that input and output tensors have data types defined, and fail if not.
kwargs – Additional arguments to onnx.helper.make_model
- Returns
A corresponding ONNX model.
- Return type
onnx.ModelProto