pytorch
7b8bd916 - fp16 -> fp32 EmbeddingBag moved into CPU impl (#47076)

Commit
4 years ago
fp16 -> fp32 EmbeddingBag moved into CPU impl (#47076) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/47076 Pull Request resolved: https://github.com/pytorch/glow/pull/5038 Eliminate double casting in glow when submitting fp16 per sample weights Test Plan: buck test glow/glow/torch_glow/tests:embedding_bag_test Due to dependency conflicts between glow and caffe2, the test has been reverted from this diff, and landed separately Reviewed By: allwu Differential Revision: D24421367 fbshipit-source-id: eb3615144a2cad3d593543428dfdec165ad301df
Author
Parents
Loading