pytorch

1
# Defined in torch/csrc/onnx/init.cpp
2

3
from enum import Enum
4

5
PRODUCER_VERSION: str
6

7
class 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

30
class OperatorExportTypes(Enum):
31
    ONNX = ...
32
    ONNX_ATEN = ...
33
    ONNX_ATEN_FALLBACK = ...
34
    ONNX_FALLTHROUGH = ...
35

36
class TrainingMode(Enum):
37
    EVAL = ...
38
    PRESERVE = ...
39
    TRAINING = ...
40