|
|
|
|
|
from enum import Enum |
|
|
|
_CAFFE2_ATEN_FALLBACK: bool |
|
PRODUCER_VERSION: str |
|
|
|
class TensorProtoDataType(Enum): |
|
UNDEFINED = ... |
|
FLOAT = ... |
|
UINT8 = ... |
|
INT8 = ... |
|
UINT16 = ... |
|
INT16 = ... |
|
INT32 = ... |
|
INT64 = ... |
|
STRING = ... |
|
BOOL = ... |
|
FLOAT16 = ... |
|
DOUBLE = ... |
|
UINT32 = ... |
|
UINT64 = ... |
|
COMPLEX64 = ... |
|
COMPLEX128 = ... |
|
BFLOAT16 = ... |
|
|
|
class OperatorExportTypes(Enum): |
|
ONNX = ... |
|
ONNX_ATEN = ... |
|
ONNX_ATEN_FALLBACK = ... |
|
ONNX_FALLTHROUGH = ... |
|
|
|
class TrainingMode(Enum): |
|
EVAL = ... |
|
PRESERVE = ... |
|
TRAINING = ... |
|
|
