Export

onnx_graphsurgeon.export_onnx(graph: Graph, do_type_check=True, **kwargs) → 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. Defaults to True.

• kwargs – Additional arguments to onnx.helper.make_model

Returns:

A corresponding ONNX model.

Return type:

onnx.ModelProto