• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# flake8: noqa: F401
2r"""Quantized Reference Modules.
3
4This module is in the process of migration to
5`torch/ao/nn/quantized/reference`, and is kept here for
6compatibility while the migration process is ongoing.
7If you are adding a new entry/functionality, please, add it to the
8appropriate file under the `torch/ao/nn/quantized/reference`,
9while adding an import statement here.
10"""
11
12from torch.ao.nn.quantized.reference.modules.rnn import (
13    GRUCell,
14    LSTM,
15    LSTMCell,
16    RNNBase,
17    RNNCell,
18    RNNCellBase,
19)
20