pytorch
a0ae416d - [quant] Support aten::embedding_bag quantization in graph mode (#43989)

Commit
4 years ago
[quant] Support aten::embedding_bag quantization in graph mode (#43989) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/43989 When we trace the model it produces aten::embedding_bag node in the graph, Add necessary passes in graph mode to help support quantizing it as well Test Plan: python test/test_quantization.py TestQuantizeDynamicJitOps.test_embedding_bag Imported from OSS Reviewed By: vkuzo Differential Revision: D23460485 fbshipit-source-id: 328c5e1816cfebb10ba951113f657665b6d17575
Author
Parents
Loading