sparse.py 524 B

12345678910111213
  1. # flake8: noqa: F401
  2. r"""Quantized Reference Modules
  3. This module is in the process of migration to
  4. `torch/ao/nn/quantized/reference`, and is kept here for
  5. compatibility while the migration process is ongoing.
  6. If you are adding a new entry/functionality, please, add it to the
  7. appropriate file under the `torch/ao/nn/quantized/reference`,
  8. while adding an import statement here.
  9. """
  10. from torch.ao.nn.quantized.reference.modules.sparse import Embedding
  11. from torch.ao.nn.quantized.reference.modules.sparse import EmbeddingBag