mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
521d1071f8
Some of the subpackages were not included in the 'torch.nn.quantized'. That would cause some specific cases fail. For example, `from torch.nn.quantized import dynamic` would work, but `import torch; torch.nn.quantized.dynamic` would fail. Fixes #ISSUE_NUMBER Pull Request resolved: https://github.com/pytorch/pytorch/pull/84141 Approved by: https://github.com/andrewor14
19 lines
366 B
Python
19 lines
366 B
Python
# flake8: noqa: F401
|
|
r"""QAT Dynamic Modules
|
|
|
|
This package is in the process of being deprecated.
|
|
Please, use `torch.ao.nn.qat.dynamic` instead.
|
|
"""
|
|
from . import dynamic # noqa: F403
|
|
from . import modules # noqa: F403
|
|
from .modules import * # noqa: F403
|
|
|
|
__all__ = [
|
|
"Linear",
|
|
"Conv1d",
|
|
"Conv2d",
|
|
"Conv3d",
|
|
"Embedding",
|
|
"EmbeddingBag",
|
|
]
|