Import

onnx_graphsurgeon.import_onnx(onnx_model: onnx.onnx_ml_pb2.ModelProto) onnx_graphsurgeon.ir.graph.Graph

Import an onnx-graphsurgeon Graph from the provided ONNX model.

Parameters

onnx_model (onnx.ModelProto) – The ONNX model.

Returns

A corresponding onnx-graphsurgeon Graph.

Return type

Graph