projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 0561e10) | patch
handle the case in acc_ops.sum when dim == 0, differentiating it from the case when...
authorEmad El-Haraty <elharaty@fb.com>
Mon, 13 Sep 2021 21:22:53 +0000 (14:22 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Mon, 13 Sep 2021 21:24:16 +0000 (14:24 -0700)
commitc99277e177cf16736262251c7e92ea5e9ba2c5c2
treea48f6dc5d784302166191719b3965c799f5f8c2dtree | snapshot
parent0561e104d97cb2abf9c833a8361e2ae5645bc2decommit | diff
handle the case in acc_ops.sum when dim == 0, differentiating it from the case when dim is None (#64869)

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

handle the case in acc_ops.sum when dim == 0, differentiating it from the case when dim is None

Reviewed By: 842974287

Differential Revision: D30872739

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