mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
3a02873183
All quantization-related modules are being migrated to `torch.ao`. This migrates the `nn.intrinsic.quantized`. Please, see the [tracker](https://github.com/pytorch/pytorch/issues/81667) for the timeline. ``` python test/test_quantization.py -- TestAOMigrationNNIntrinsic ``` Internal: ``` buck2 test @mode/dev-nosan //caffe2/test:quantization -- TestAOMigrationNNIntrinsic ``` Differential Revision: [D39425515](https://our.internmc.facebook.com/intern/diff/D39425515/) Differential Revision: [D39425515](https://our.internmc.facebook.com/intern/diff/D39425515) Pull Request resolved: https://github.com/pytorch/pytorch/pull/86172 Approved by: https://github.com/jerryzh168
36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
from torch.ao.nn.intrinsic import ConvBn1d
|
|
from torch.ao.nn.intrinsic import ConvBn2d
|
|
from torch.ao.nn.intrinsic import ConvBn3d
|
|
from torch.ao.nn.intrinsic import ConvBnReLU1d
|
|
from torch.ao.nn.intrinsic import ConvBnReLU2d
|
|
from torch.ao.nn.intrinsic import ConvBnReLU3d
|
|
from torch.ao.nn.intrinsic import ConvReLU1d
|
|
from torch.ao.nn.intrinsic import ConvReLU2d
|
|
from torch.ao.nn.intrinsic import ConvReLU3d
|
|
from torch.ao.nn.intrinsic import LinearReLU
|
|
from torch.ao.nn.intrinsic import BNReLU2d
|
|
from torch.ao.nn.intrinsic import BNReLU3d
|
|
from torch.ao.nn.intrinsic import LinearBn1d
|
|
from torch.ao.nn.intrinsic.modules.fused import _FusedModule # noqa: F401
|
|
|
|
# Include the subpackages in case user imports from it directly
|
|
from . import modules # noqa: F401
|
|
from . import qat # noqa: F401
|
|
from . import quantized # noqa: F401
|
|
|
|
__all__ = [
|
|
'ConvBn1d',
|
|
'ConvBn2d',
|
|
'ConvBn3d',
|
|
'ConvBnReLU1d',
|
|
'ConvBnReLU2d',
|
|
'ConvBnReLU3d',
|
|
'ConvReLU1d',
|
|
'ConvReLU2d',
|
|
'ConvReLU3d',
|
|
'LinearReLU',
|
|
'BNReLU2d',
|
|
'BNReLU3d',
|
|
'LinearBn1d',
|
|
]
|