projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 2223737) | patch
Adding support for lowering 4Bit EmbeddingBag Operator (#5806)
authorProtonu Basu <protonu@fb.com>
Wed, 8 Sep 2021 14:11:38 +0000 (07:11 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Wed, 8 Sep 2021 14:13:16 +0000 (07:13 -0700)
commitb616132403697a043fd9691693e40f407e77063a
treeb7a7d0228fa41621f63760bd91f63872a27e0c1ctree | snapshot
parent2223737da91454336ccf880b75718db7369cdc8fcommit | diff
Adding support for lowering 4Bit EmbeddingBag Operator (#5806)

Summary:
Pull Request resolved: https://github.com/pytorch/glow/pull/5806

Pull Request resolved: https://github.com/pytorch/pytorch/pull/64001

Add 4 bit embeddingbag operator in  acc_ops.

Test Plan: Let CI run.

Reviewed By: jfix71

Differential Revision: D30532824

fbshipit-source-id: bf476c9710477792aae202dacf64e23539c33bd9
torch/fx/experimental/fx_acc/acc_ops.py diff | blob | history
torch/fx/experimental/graph_manipulation.py diff | blob | history
Domain: Machine Learning / ML Framework;