projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 2a2007e) | patch
EmbeddingBag w/ per_sample_weights CPU backward (#18799)
authorRichard Zou <zou3519@gmail.com>
Wed, 10 Apr 2019 01:08:59 +0000 (18:08 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 10 Apr 2019 01:12:59 +0000 (18:12 -0700)
commit0397d7c0c8501aaa8c4fe2a4857f43213adbfd5d
treea10b82e2ed06d4a4468ff99dfb275b5d9c723585tree | snapshot
parent2a2007e5aca575eb495e42dbf4404451a351eecacommit | diff
EmbeddingBag w/ per_sample_weights CPU backward (#18799)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18799
ghimport-source-id: 58a6f629e890449013f24a9b6282664ca2a1e3ba

Reviewed By: cpuhrsch

Differential Revision: D14851417

Pulled By: zou3519

fbshipit-source-id: c36b9d469989354bf6cef1c2c3dc4f13e7cb1a25
aten/src/ATen/native/EmbeddingBag.cpp diff | blob | history
aten/src/ATen/native/cuda/EmbeddingBag.cu diff | blob | history
aten/src/ATen/native/native_functions.yaml diff | blob | history
test/test_nn.py diff | blob | history
Domain: Machine Learning / ML Framework;