1# Defined in torch/csrc/onnx/init.cpp 2 3from enum import Enum 4 5PRODUCER_VERSION: str 6 7class TensorProtoDataType(Enum): 8 UNDEFINED = ... 9 FLOAT = ... 10 UINT8 = ... 11 INT8 = ... 12 UINT16 = ... 13 INT16 = ... 14 INT32 = ... 15 INT64 = ... 16 STRING = ... 17 BOOL = ... 18 FLOAT16 = ... 19 DOUBLE = ... 20 UINT32 = ... 21 UINT64 = ... 22 COMPLEX64 = ... 23 COMPLEX128 = ... 24 BFLOAT16 = ... 25 FLOAT8E5M2 = ... 26 FLOAT8E4M3FN = ... 27 FLOAT8E5M2FNUZ = ... 28 FLOAT8E4M3FNUZ = ... 29 30class OperatorExportTypes(Enum): 31 ONNX = ... 32 ONNX_ATEN = ... 33 ONNX_ATEN_FALLBACK = ... 34 ONNX_FALLTHROUGH = ... 35 36class TrainingMode(Enum): 37 EVAL = ... 38 PRESERVE = ... 39 TRAINING = ... 40