pytorch
343cb276 - [pytorch] Add broadcasting support to add_relu kernel (#61584)

Commit
4 years ago
[pytorch] Add broadcasting support to add_relu kernel (#61584) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/61584 add_relu is not working with broadcasting. This registers a scalar version of add_relu in native_functions that casts to tensor before calling the regular function. TensorIterator handles broadcasting analogously to existing add. ghstack-source-id: 133480068 Test Plan: python3 test/test_nn.py TestAddRelu Reviewed By: kimishpatel Differential Revision: D29641768 fbshipit-source-id: 1b0ecfdb7eaf44afed83c9e9e74160493c048cbc
Author
John Shen
Parents
Loading