Home Explore Help
Sign In
SRI-DYZBC2
/
Vehicle-cpp
2
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Tree: e6b2cb2e99
Branches Tags
Vehicle-cpp
/
hkpc
/
software
/
torch
/
quantization
/
fx
/
quantization_patterns.py

quantization_patterns.py 2.0 KB
History Raw

123456789101112131415161718192021222324252627282930313233343536373839 
  1. # flake8: noqa: F401
    2. 

  2. r"""
    3. 

  3. This file is in the process of migration to `torch/ao/quantization`, and
    4. 

  4. is kept here for compatibility while the migration process is ongoing.
    5. 

  5. If you are adding a new entry/functionality, please, add it to the
    6. 

  6. appropriate files under `torch/ao/quantization/fx/`, while adding an import statement
    7. 

  7. here.
    8. 

  8. """
    9. 

  9. from torch.ao.quantization.fx.quantize_handler import (
    10. 

  10.     QuantizeHandler,
    11. 

  11.     BinaryOpQuantizeHandler,
    12. 

  12.     CatQuantizeHandler,
    13. 

  13.     ConvReluQuantizeHandler,
    14. 

  14.     LinearReLUQuantizeHandler,
    15. 

  15.     BatchNormQuantizeHandler,
    16. 

  16.     EmbeddingQuantizeHandler,
    17. 

  17.     RNNDynamicQuantizeHandler,
    18. 

  18.     DefaultNodeQuantizeHandler,
    19. 

  19.     FixedQParamsOpQuantizeHandler,
    20. 

  20.     CopyNodeQuantizeHandler,
    21. 

  21.     CustomModuleQuantizeHandler,
    22. 

  22.     GeneralTensorShapeOpQuantizeHandler,
    23. 

  23.     StandaloneModuleQuantizeHandler
    24. 

  24. )
    25. 

  25. 

  26. QuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns"
    27. 

  27. BinaryOpQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns"
    28. 

  28. CatQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns"
    29. 

  29. ConvReluQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns"
    30. 

  30. LinearReLUQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns"
    31. 

  31. BatchNormQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns"
    32. 

  32. EmbeddingQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns"
    33. 

  33. RNNDynamicQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns"
    34. 

  34. DefaultNodeQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns"
    35. 

  35. FixedQParamsOpQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns"
    36. 

  36. CopyNodeQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns"
    37. 

  37. CustomModuleQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns"
    38. 

  38. GeneralTensorShapeOpQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns"
    39. 

  39. StandaloneModuleQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns"
    40.