[ONNX] Export embedding_bag (#41234)
Summary:
Enable export of embedding_bag op to ONNX
Pull Request resolved: https://github.com/pytorch/pytorch/pull/41234
Reviewed By: houseroad
Differential Revision: D22567470
Pulled By: bzinodev
fbshipit-source-id: 2fcf74e54f3a9dee4588d7877a4ac9eb6c2a3629