File size: 331 Bytes
59f1501 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
from enum import IntEnum
# 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(IntEnum):
rnn_relu = ...
rnn_tanh = ...
lstm = ...
gru = ...
|