/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/_cudnn.pyi (378B)
from enum import Enum from torch.types import Tuple, Number, _bool # Defined in torch/csrc/cuda/shared/cudnn.cpp is_cuda: _bool def getRuntimeVersion() -> Tuple[int, int, int]: ... def getCompileVersion() -> Tuple[int, int, int]: ... def getVersionInt() -> int: ... class RNNMode(int, Enum): value: int rnn_relu = ... rnn_tanh = ... lstm = ... gru = ...