pytorch
a788c2d7 - [nnc] Remove output_args from ReduceOp (#52187)

Commit
4 years ago
[nnc] Remove output_args from ReduceOp (#52187) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/52187 ReduceOp doesn't need to track the indices that its result will be written into. ghstack-source-id: 121813075 Test Plan: test_tensorexpr, tensorexpr_bench Imported from OSS Reviewed By: ZolotukhinM Differential Revision: D26420575 fbshipit-source-id: 7afcfa611515334e36de8039722011687f3b61e4
Author
Parents
Loading