pytorch
d32a7621 - [quant][ao_migration] `torch.nn.quantized.dynamic` → `torch.ao.nn.quantized.dynamic` (#78714)

Comment changes are shownComment changes are hidden
Commit
2 years ago
[quant][ao_migration] `torch.nn.quantized.dynamic` → `torch.ao.nn.quantized.dynamic` (#78714) Context: In order to avoid the cluttering of the `torch.nn` namespace the quantized modules namespace is moved to `torch.ao.nn`. The list of the `nn.quantized` files that are being migrated: - [ ] `torch.nn.quantized` → `torch.ao.nn.quantized` - [X] `torch.nn.quantized.functional` → `torch.ao.nn.quantized.functional` - [X] `torch.nn.quantized.modules` → `torch.ao.nn.quantized.modules` - [X] [Current PR] `torch.nn.quantized.dynamic` → `torch.ao.nn.quantized.dynamic` - [ ] `torch.nn.quantized._reference` → `torch.ao.nn.quantized._reference` - [ ] `torch.nn.quantizable` → `torch.ao.nn.quantizable` - [ ] `torch.nn.qat` → `torch.ao.nn.qat` - [ ] `torch.nn.qat.modules` → `torch.ao.nn.qat.modules` - [ ] `torch.nn.qat.dynamic` → `torch.ao.nn.qat.dynamic` - [ ] `torch.nn.intrinsic` → `torch.ao.nn.intrinsic` - [ ] `torch.nn.intrinsic.modules` → `torch.ao.nn.intrinsic.modules` - [ ] `torch.nn.intrinsic.qat` → `torch.ao.nn.intrinsic.qat` - [ ] `torch.nn.intrinsic.quantized` → `torch.ao.nn.intrinsic.quantized` - [ ] `torch.nn.intrinsic.quantized.modules` → `torch.ao.nn.intrinsic.quantized.modules` - [ ] `torch.nn.intrinsic.quantized.dynamic` → `torch.ao.nn.intrinsic.quantized.dynamic` Majority of the files are just moved to the new location. However, specific files need to be double checked: - [Documentation](docs/source/quantization-support.rst) @vkuzo - [Public API test list](test/allowlist_for_publicAPI.json) @peterbell10 - [BC test](test/quantization/bc/test_backward_compatibility.py) @vkuzo - [IR emitter](torch/csrc/jit/frontend/ir_emitter.cpp) @jamesr66a - [JIT serialization](torch/csrc/jit/serialization/import_source.cpp) @IvanKobzarev @jamesr66a Differential Revision: [D36860660](https://our.internmc.facebook.com/intern/diff/D36860660/) **NOTE FOR REVIEWERS**: This PR has internal Facebook specific changes or comments, please review them on [Phabricator](https://our.internmc.facebook.com/intern/diff/D36860660/)! Differential Revision: [D36860660](https://our.internmc.facebook.com/intern/diff/D36860660) Pull Request resolved: https://github.com/pytorch/pytorch/pull/78714 Approved by: https://github.com/jerryzh168
Author
zaf
Committer
Parents
  • benchmarks/operator_benchmark/pt
    • File
      qlinear_test.py
  • docs/source
    • File
      quantization-support.rst
    • File
      quantization.rst
  • test
    • File
      allowlist_for_publicAPI.json
    • quantization
      • ao_migration
        • File
          test_ao_migration.py
      • bc
        • File
          test_backward_compatibility.py
      • core
        • File
          test_quantized_module.py
      • eager
        • File
          test_quantize_eager_ptq.py
        • File
          test_quantize_eager_qat.py
      • fx
        • File
          test_quantize_fx.py
  • torch
    • ao
      • nn/quantized/dynamic
        • File
          __init__.py
        • modules
          • File
            __init__.py
          • File
            conv.py
          • File
            linear.py
          • File
            rnn.py
      • ns
        • File
          _numeric_suite.py
        • fx
          • File
            mappings.py
          • File
            weight_utils.py
      • quantization
        • fx
          • File
            _lower_to_native_backend.py
        • File
          quantization_mappings.py
    • csrc/jit/serialization
      • File
        import_source.cpp
    • jit
      • File
        quantized.py
    • nn
      • intrinsic/quantized/dynamic/modules
        • File
          linear_relu.py
      • quantized/dynamic
        • File
          __init__.py
        • modules
          • File
            __init__.py
          • File
            conv.py
          • File
            linear.py
          • File
            rnn.py
    • testing/_internal
      • File
        common_quantization.py