Lines Matching full:onnx

1 torch.onnx
7 `Open Neural Network eXchange (ONNX) <https://onnx.ai/>`_ is an open standard
8 format for representing machine learning models. The ``torch.onnx`` module captures the computation…
10 `ONNX graph <https://github.com/onnx/onnx/blob/main/docs/IR.md>`_.
13 `runtimes that support ONNX <https://onnx.ai/supported-tools.html#deployModel>`_, including
14 Microsoft's `ONNX Runtime <https://www.onnxruntime.ai>`_.
16 **There are two flavors of ONNX exporter API that you can use, as listed below.**
17 Both can be called through function :func:`torch.onnx.export`.
36     torch.onnx.export(
39         "my_model.onnx",        # filename of the ONNX model
40         input_names=["input"],  # Rename inputs for the ONNX model
46 TorchDynamo-based ONNX Exporter
49 *The TorchDynamo-based ONNX exporter is the newest (and Beta) exporter for PyTorch 2.1 and newer*
53 ONNX graph.
58 :doc:`Learn more about the TorchDynamo-based ONNX Exporter <onnx_dynamo>`
60 TorchScript-based ONNX Exporter
63 *The TorchScript-based ONNX exporter is available since PyTorch 1.2.0*
78 :doc:`Learn more about the TorchScript-based ONNX Exporter <onnx_torchscript>`
83 The ONNX exporter is a community project and we welcome contributions. We follow the
85 …in reading our `development wiki <https://github.com/pytorch/pytorch/wiki/PyTorch-ONNX-exporter>`_.
96 .. py:module:: torch.onnx.errors
97 .. py:module:: torch.onnx.operators
98 .. py:module:: torch.onnx.symbolic_caffe2
99 .. py:module:: torch.onnx.symbolic_helper
100 .. py:module:: torch.onnx.symbolic_opset10
101 .. py:module:: torch.onnx.symbolic_opset11
102 .. py:module:: torch.onnx.symbolic_opset13
103 .. py:module:: torch.onnx.symbolic_opset14
104 .. py:module:: torch.onnx.symbolic_opset15
105 .. py:module:: torch.onnx.symbolic_opset16
106 .. py:module:: torch.onnx.symbolic_opset17
107 .. py:module:: torch.onnx.symbolic_opset18
108 .. py:module:: torch.onnx.symbolic_opset19
109 .. py:module:: torch.onnx.symbolic_opset20
110 .. py:module:: torch.onnx.symbolic_opset7
111 .. py:module:: torch.onnx.symbolic_opset8
112 .. py:module:: torch.onnx.symbolic_opset9
113 .. py:module:: torch.onnx.utils
114 .. py:module:: torch.onnx.verification
115 .. py:module:: torch.onnx.symbolic_opset12