pytorch
0cd155b0 - [reland][quant][pt2e] Annotate GRU module (#103358) (#103526)

Commit
1 year ago
[reland][quant][pt2e] Annotate GRU module (#103358) (#103526) Summary: att, we use module partition API to identify the GRU submodule and annotate all necessary patterns Test Plan: buck2 test mode/opt caffe2/test:quantization_pt2e -- 'caffe2/test:quantization_pt2e' Differential Revision: D46689428 Pull Request resolved: https://github.com/pytorch/pytorch/pull/103526 Approved by: https://github.com/andrewor14
Author
Committer
Parents
Loading