pytorch
df47fa5b - Using meta checks for unary `torch.all` and `torch.any`. (#60362)

Commit
3 years ago
Using meta checks for unary `torch.all` and `torch.any`. (#60362) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/60362 This PR makes use of the newly implemented unified `at::meta::check_reduction` for validating the inputs and configuring its `TensorIterator`. This PR was openned so as to solve the CI failures in main when merging: #59371 #59372 #59373 #59937 #59938. Test Plan: Imported from OSS Reviewed By: albanD Differential Revision: D29265858 Pulled By: ezyang fbshipit-source-id: e8961b7da65a31acfed5ac3f5c1f5985ae81ec37
Author
Parents
Loading