normalization.py 796 B

1234567891011121314151617
  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__ = ['LayerNorm', 'GroupNorm', 'InstanceNorm1d', 'InstanceNorm2d', 'InstanceNorm3d']
  10. from torch.ao.nn.quantized.modules.normalization import LayerNorm
  11. from torch.ao.nn.quantized.modules.normalization import GroupNorm
  12. from torch.ao.nn.quantized.modules.normalization import InstanceNorm1d
  13. from torch.ao.nn.quantized.modules.normalization import InstanceNorm2d
  14. from torch.ao.nn.quantized.modules.normalization import InstanceNorm3d