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