projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 92a154a) | patch
[pytorch][quant][oss] Support 2-bit embedding_bag op "embedding_bag_2bit_rowwise_offs...
authorShijun Kong <shijunk@fb.com>
Thu, 26 Aug 2021 23:06:17 +0000 (16:06 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Thu, 26 Aug 2021 23:09:35 +0000 (16:09 -0700)
commitdfa35ab3e710848353aa1d313c5d9127ed2ef745
tree97b8a8d3cd11da422c9f429a5a7eb06050ad250atree | snapshot
parent92a154aa29186afea961a6fe491721229543535fcommit | diff
[pytorch][quant][oss] Support 2-bit embedding_bag op "embedding_bag_2bit_rowwise_offsets" (#63658)

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

Support 2-bit embedding_bag op "embedding_bag_2bit_rowwise_offsets"

Reviewed By: jingsh, supriyar

Differential Revision: D30454994

fbshipit-source-id: 7aa7bfe405c2ffff639d5658a35181036e162dc9
aten/src/ATen/native/quantized/cpu/qembeddingbag.cpp diff | blob | history
aten/src/ATen/native/quantized/library.cpp diff | blob | history
test/quantization/core/test_quantized_op.py diff | blob | history
Domain: Machine Learning / ML Framework;