| 123456789101112131415161718192021222324252627282930313233343536373839 |
- # 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 = ...
|
