projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 1280363) | patch
Port `prod` kernel to structured kernels. (#62024)
authorYukio Siraichi <yukio.siraichi@gmail.com>
Fri, 13 Aug 2021 15:20:19 +0000 (08:20 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Fri, 13 Aug 2021 15:27:46 +0000 (08:27 -0700)
commit07bb6e4fd08b35a9080d0232571b61555f422cdb
treeb6210d4cd4005bb6ad1ec7e33e904114bec5963btree | snapshot
parent1280363badddd622481695ac677dc8b75cad6a48commit | diff
Port `prod` kernel to structured kernels. (#62024)

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

Tracking issue: #55070

In this PR, I also broke down the meta functions of other reduction kernels (e.g. `all`,
`argmax`, `sum`) into the composition of common patterns.

Test Plan: Imported from OSS

Reviewed By: ejguan

Differential Revision: D29847122

Pulled By: ezyang

fbshipit-source-id: a6680a6cf6e59bb46b8ffe7bf2a3a611d6e0fd14
aten/src/ATen/native/ReduceOps.cpp diff | blob | history
aten/src/ATen/native/native_functions.yaml diff | blob | history
Domain: Machine Learning / ML Framework;