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