• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# flake8: noqa: F401
2r"""Quantized Dynamic Modules.
3
4This file is in the process of migration to `torch/ao/nn/quantized/dynamic`,
5and is kept here for compatibility while the migration process is ongoing.
6If you are adding a new entry/functionality, please, add it to the
7appropriate file under the `torch/ao/nn/quantized/dynamic/modules`,
8while adding an import statement here.
9"""
10
11from torch.ao.nn.quantized.dynamic.modules.rnn import (
12    GRU,
13    GRUCell,
14    LSTM,
15    LSTMCell,
16    pack_weight_bias,
17    PackedParameter,
18    RNNBase,
19    RNNCell,
20    RNNCellBase,
21)
22
23
24__all__ = [
25    "pack_weight_bias",
26    "PackedParameter",
27    "RNNBase",
28    "LSTM",
29    "GRU",
30    "RNNCellBase",
31    "RNNCell",
32    "LSTMCell",
33    "GRUCell",
34]
35