[quant] Support set API for EmbeddingBag quantization (#43433)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/43433
Add support for torch.quint8 dtype
Test Plan: Imported from OSS
Reviewed By: radkris-git
Differential Revision: D23277002
fbshipit-source-id: 4204bc62f124b4fd481aaa6aa47b9437978c43ee