1# flake8: noqa: F401 2r"""QAT Modules. 3 4This file is in the process of migration to `torch/ao/nn/qat`, and 5is 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/qat/modules`, 8while adding an import statement here. 9""" 10 11from torch.ao.nn.qat.modules.embedding_ops import Embedding, EmbeddingBag 12 13 14__all__ = ["Embedding", "EmbeddingBag"] 15