Save a heap allocation in embedding_bag_{byte,nbit}_impl
vector that could be a stack array
Differential Revision: [D36181474](https://our.internmc.facebook.com/intern/diff/D36181474/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/77564
Approved by: https://github.com/jianyuh
