/usr/local/lib64/python3.6/site-packages/torch/_C
NameSizeModeActions
_autograd.pyi27440644editdlrm
_cudnn.pyi3780644editdlrm
_distributed_autograd.pyi9080644editdlrm
_distributed_c10d.pyi93760644editdlrm
_distributed_rpc.pyi56300644editdlrm
_distributed_rpc_testing.pyi11360644editdlrm
_functions.pyi3450644editdlrm
_nn.pyi20500644editdlrm
_nvtx.pyi1500644editdlrm
_onnx.pyi6460644editdlrm
_VariableFunctions.pyi1410260644editdlrm
__init__.pyi1039060644editdlrm
Edit: /usr/local/lib64/python3.6/site-packages/torch/_C/_onnx.pyi (646B)
# Defined in torch/csrc/onnx/init.cpp from enum import Enum PYTORCH_ONNX_CAFFE2_BUNDLE: bool IR_VERSION: int PRODUCER_VERSION: str class TensorProtoDataType(Enum): UNDEFINED = ... FLOAT = ... UINT8 = ... INT8 = ... UINT16 = ... INT16 = ... INT32 = ... INT64 = ... STRING = ... BOOL = ... FLOAT16 = ... DOUBLE = ... UINT32 = ... UINT64 = ... COMPLEX64 = ... COMPLEX128 = ... class OperatorExportTypes(Enum): ONNX = ... ONNX_ATEN = ... ONNX_ATEN_FALLBACK = ... ONNX_FALLTHROUGH = ... class TrainingMode(Enum): EVAL = ... PRESERVE = ... TRAINING = ...