conv.py 882 B

123456789101112131415161718192021
  1. # flake8: noqa: F401
  2. r"""Quantized Modules
  3. This file is in the process of migration to `torch/ao/nn/quantized`, and
  4. is kept here for compatibility while the migration process is ongoing.
  5. If you are adding a new entry/functionality, please, add it to the
  6. appropriate file under the `torch/ao/nn/quantized/modules`,
  7. while adding an import statement here.
  8. """
  9. __all__ = ['Conv1d', 'Conv2d', 'Conv3d', 'ConvTranspose1d', 'ConvTranspose2d', 'ConvTranspose3d']
  10. from torch.ao.nn.quantized.modules.conv import _reverse_repeat_padding
  11. from torch.ao.nn.quantized.modules.conv import Conv1d
  12. from torch.ao.nn.quantized.modules.conv import Conv2d
  13. from torch.ao.nn.quantized.modules.conv import Conv3d
  14. from torch.ao.nn.quantized.modules.conv import ConvTranspose1d
  15. from torch.ao.nn.quantized.modules.conv import ConvTranspose2d
  16. from torch.ao.nn.quantized.modules.conv import ConvTranspose3d