1from enum import Enum 2 3from torch.types import _bool, Tuple 4 5# Defined in torch/csrc/cuda/shared/cudnn.cpp 6is_cuda: _bool 7 8def getRuntimeVersion() -> Tuple[int, int, int]: ... 9def getCompileVersion() -> Tuple[int, int, int]: ... 10def getVersionInt() -> int: ... 11 12class RNNMode(int, Enum): 13 value: int 14 rnn_relu = ... 15 rnn_tanh = ... 16 lstm = ... 17 gru = ... 18