pytorch

ee48bd08 - Support mix of int32 and int64 offsets/indices for EmbeddingBag and its variants (#55189)

Support mix of int32 and int64 offsets/indices for EmbeddingBag and its variants (#55189) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/55189 Currently EmbeddingBag and it variants support either int32 or int64 indices/offsets. We have use cases where there are mix of int32 and int64 indices which are not supported yet. To avoid introducing too many branches we could simply cast offsets type to indices type when they are not the same. Test Plan: unit tests Reviewed By: allwu Differential Revision: D27482738 fbshipit-source-id: deeadd391d49ff65d17d016092df1839b82806cc