Classes | |
class | OnnxExporter |
Functions | |
int | dtype_to_onnx (np.dtype dtype) |
"onnx.ModelProto" | export_onnx (Graph graph, do_type_check=True, **kwargs) |
int onnx_graphsurgeon.exporters.onnx_exporter.dtype_to_onnx | ( | np.dtype | dtype | ) |
"onnx.ModelProto" onnx_graphsurgeon.exporters.onnx_exporter.export_onnx | ( | Graph | graph, |
do_type_check = True , |
|||
** | kwargs | ||
) |
Exports an onnx-graphsurgeon Graph to an ONNX model. Args: 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: onnx.ModelProto: A corresponding ONNX model.