| 1234567891011121314 |
- 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 = ...
|
