| # Defined in torch/csrc/onnx/init.cpp | |
| from enum import Enum | |
| PRODUCER_VERSION: str | |
| class TensorProtoDataType(Enum): | |
| UNDEFINED = ... | |
| FLOAT = ... | |
| UINT8 = ... | |
| INT8 = ... | |
| UINT16 = ... | |
| INT16 = ... | |
| INT32 = ... | |
| INT64 = ... | |
| STRING = ... | |
| BOOL = ... | |
| FLOAT16 = ... | |
| DOUBLE = ... | |
| UINT32 = ... | |
| UINT64 = ... | |
| COMPLEX64 = ... | |
| COMPLEX128 = ... | |
| BFLOAT16 = ... | |
| FLOAT8E5M2 = ... | |
| FLOAT8E4M3FN = ... | |
| FLOAT8E5M2FNUZ = ... | |
| FLOAT8E4M3FNUZ = ... | |
| class OperatorExportTypes(Enum): | |
| ONNX = ... | |
| ONNX_ATEN = ... | |
| ONNX_ATEN_FALLBACK = ... | |
| ONNX_FALLTHROUGH = ... | |
| class TrainingMode(Enum): | |
| EVAL = ... | |
| PRESERVE = ... | |
| TRAINING = ... | |