Kano001's picture
Upload 5252 files
c61ccee verified
raw
history blame
387 Bytes
from enum import Enum
from torch.types import _bool, Tuple
# 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 = ...