pytorch
553bb01d - [quant][pt2e][refactor] Remove extra arguments of _maybe_insert_observers_before_graph_output (#98029)

Comment changes are shownComment changes are hidden
Commit
2 years ago
[quant][pt2e][refactor] Remove extra arguments of _maybe_insert_observers_before_graph_output (#98029) Summary: This PR allows _maybe_insert_observers_before_graph_output to be reused by pt2e flow Test Plan: python test/test_quantization.py TestQuantizeFx python test/test_quantization.py TestQuantizeFxOps python test/test_quantization.py TestQuantizeFxModels Reviewers: Subscribers: Tasks: Tags: Pull Request resolved: https://github.com/pytorch/pytorch/pull/98029 Approved by: https://github.com/vkuzo
Author
Committer
Parents
  • torch/ao/quantization/fx
    • File
      prepare.py