pytorch
57af1ec1 - observers: use torch.all to check for valid min and max values (#43151)

Commit
4 years ago
observers: use torch.all to check for valid min and max values (#43151) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/43151 Using `torch.all` instead of `torch.sum` and length check. It's unclear whether the increase in perf (~5% for small inputs) is real, but should be a net benefit, especially for larger channel inputs. Test Plan: Imported from OSS Reviewed By: jerryzh168 Differential Revision: D23170426 fbshipit-source-id: ee5c25eb93cee1430661128ac9458a9c525df8e5
Author
Parents
Loading