pytorch
4484c045 - [quant][embdding qat] Add FX support for QAT EmbeddingBag (#68121)

Commit
3 years ago
[quant][embdding qat] Add FX support for QAT EmbeddingBag (#68121) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/68121 Add FX support for QAT EmbeddingBag operator, previously only eager mode support. Test Plan: pytest test/quantization/fx/test_quantize_fx.py -v -k "test_qat_embeddingbag_linear" Imported from OSS Reviewed By: supriyar Differential Revision: D32318435 fbshipit-source-id: 8b5d1a5d5422972c49676f9e470d5fbe29dd503b
Author
Parents
Loading