Spaces:
Running
Running
# Defined in torch/csrc/onnx/init.cpp | |
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 = ... | |
FLOAT8E5M2 = ... | |
FLOAT8E4M3FN = ... | |
FLOAT8E5M2FNUZ = ... | |
FLOAT8E4M3FNUZ = ... | |
class OperatorExportTypes(Enum): | |
ONNX = ... | |
ONNX_ATEN = ... | |
ONNX_ATEN_FALLBACK = ... | |
ONNX_FALLTHROUGH = ... | |
class TrainingMode(Enum): | |
EVAL = ... | |
PRESERVE = ... | |
TRAINING = ... | |