projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 81764d1) | patch
[pytorch] add per_sample_weights support for embedding_bag_4bit_rowwise_offsets ...
authorPaul Johnson <johnsonpaul@fb.com>
Fri, 27 Aug 2021 00:28:35 +0000 (17:28 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Fri, 27 Aug 2021 00:31:45 +0000 (17:31 -0700)
commit0c9dce90ed6a12d81b0e769b76d6b0c282326823
treee601ea89d7d72ca57342bd6e96d6530c8f59c536tree | snapshot
parent81764d1153c607e324390dcac107ea0970ba668ccommit | diff
[pytorch] add per_sample_weights support for embedding_bag_4bit_rowwise_offsets (#63605)

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

Reviewed By: houseroad

Differential Revision: D30434664

fbshipit-source-id: eb4cbae3c705f9dec5c073a56f0f23daee353bc1
aten/src/ATen/native/quantized/cuda/embedding_bag.cu diff | blob | history
Domain: Machine Learning / ML Framework;