pytorch

d8c3b2b1 - [quant][pyper] Add support for pruned weights in embedding_bag_byte lookup (#47329)

[quant][pyper] Add support for pruned weights in embedding_bag_byte lookup (#47329) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/47329 Supports pruned weights along with mapping for the compressed indices Test Plan: python test/test_quantization.py TestQuantizedEmbeddingOps Imported from OSS Reviewed By: qizzzh Differential Revision: D24719909 fbshipit-source-id: f998f4039e84bbe1886e492a3bff6aa5f56b6b0f