mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
b8194268a6
This PR removes unused `noqa` suppressions in Python code. Pull Request resolved: https://github.com/pytorch/pytorch/pull/164106 Approved by: https://github.com/albanD
40 lines
833 B
Python
40 lines
833 B
Python
r"""Intrinsic QAT Modules.
|
|
|
|
This file is in the process of migration to `torch/ao/nn/intrinsic/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/intrinsic/qat/modules`,
|
|
while adding an import statement here.
|
|
"""
|
|
|
|
from torch.ao.nn.intrinsic.qat import (
|
|
ConvBn1d,
|
|
ConvBn2d,
|
|
ConvBn3d,
|
|
ConvBnReLU1d,
|
|
ConvBnReLU2d,
|
|
ConvBnReLU3d,
|
|
ConvReLU1d,
|
|
ConvReLU2d,
|
|
ConvReLU3d,
|
|
freeze_bn_stats,
|
|
update_bn_stats,
|
|
)
|
|
|
|
|
|
__all__ = [
|
|
# Modules
|
|
"ConvBn1d",
|
|
"ConvBnReLU1d",
|
|
"ConvReLU1d",
|
|
"ConvBn2d",
|
|
"ConvBnReLU2d",
|
|
"ConvReLU2d",
|
|
"ConvBn3d",
|
|
"ConvBnReLU3d",
|
|
"ConvReLU3d",
|
|
# Utilities
|
|
"freeze_bn_stats",
|
|
"update_bn_stats",
|
|
]
|