projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 54cdf65) | patch
Port `all` and `any` full reductions to structured kernels. (#64642)
authorYukio Siraichi <yukio.siraichi@gmail.com>
Wed, 15 Sep 2021 18:05:14 +0000 (11:05 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Wed, 15 Sep 2021 18:06:47 +0000 (11:06 -0700)
commit54d060a8c91c27167925b380698331850352c841
treed36da676d919b4868342f220f86289bd0db6f6a8tree | snapshot
parent54cdf651fd6432f1a33de2e2b2b1724ac2671cddcommit | diff
Port `all` and `any` full reductions to structured kernels. (#64642)

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

Tracking issue: #55070

This PR creates out overloads for both `all` and `any` kernels (full reduction overload),
and ports them to structured kernels.

Test Plan: Imported from OSS

Reviewed By: ngimel

Differential Revision: D30867354

Pulled By: ezyang

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