Viewing File: /home/ubuntu/combine_ai/combine/lib/python3.10/site-packages/torch/nn/qat/modules/embedding_ops.py

# flake8: noqa: F401
r"""QAT Modules.

This file is in the process of migration to `torch/ao/nn/qat`, and
is kept here for compatibility while the migration process is ongoing.
If you are adding a new entry/functionality, please, add it to the
appropriate file under the `torch/ao/nn/qat/modules`,
while adding an import statement here.
"""

__all__ = ['Embedding', 'EmbeddingBag']

from torch.ao.nn.qat.modules.embedding_ops import Embedding
from torch.ao.nn.qat.modules.embedding_ops import EmbeddingBag
Back to Directory File Manager