pytorch

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

Support mix of int32 and int64 offsets/indices for EmbeddingBag and its variants (#53655) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/53655 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: qizzzh Differential Revision: D26820202 fbshipit-source-id: 3e8f09523329ea12393ea92ee9a6315aa40a0b7f