1# flake8: noqa: F401 2r""" 3This file is in the process of migration to `torch/ao/quantization`, and 4is kept here for compatibility while the migration process is ongoing. 5If you are adding a new entry/functionality, please, add it to the 6appropriate files under `torch/ao/quantization/fx/`, while adding an import statement 7here. 8""" 9from torch.ao.quantization.fx.quantize_handler import ( 10 BatchNormQuantizeHandler, 11 BinaryOpQuantizeHandler, 12 CatQuantizeHandler, 13 ConvReluQuantizeHandler, 14 CopyNodeQuantizeHandler, 15 CustomModuleQuantizeHandler, 16 DefaultNodeQuantizeHandler, 17 EmbeddingQuantizeHandler, 18 FixedQParamsOpQuantizeHandler, 19 GeneralTensorShapeOpQuantizeHandler, 20 LinearReLUQuantizeHandler, 21 QuantizeHandler, 22 RNNDynamicQuantizeHandler, 23 StandaloneModuleQuantizeHandler, 24) 25 26 27QuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns" 28BinaryOpQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns" 29CatQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns" 30ConvReluQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns" 31LinearReLUQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns" 32BatchNormQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns" 33EmbeddingQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns" 34RNNDynamicQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns" 35DefaultNodeQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns" 36FixedQParamsOpQuantizeHandler.__module__ = ( 37 "torch.ao.quantization.fx.quantization_patterns" 38) 39CopyNodeQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns" 40CustomModuleQuantizeHandler.__module__ = ( 41 "torch.ao.quantization.fx.quantization_patterns" 42) 43GeneralTensorShapeOpQuantizeHandler.__module__ = ( 44 "torch.ao.quantization.fx.quantization_patterns" 45) 46StandaloneModuleQuantizeHandler.__module__ = ( 47 "torch.ao.quantization.fx.quantization_patterns" 48) 49