diff options
author | 2016-08-26 17:21:44 -0800 | |
---|---|---|
committer | 2016-08-26 18:32:27 -0700 | |
commit | f2f582b3c00744c5e8857a309d38b374a5bd60fe (patch) | |
tree | 476640eca67e44817f129878451c00354c1966fc /tensorflow/core/kernels/cwise_op_gpu_sqrt.cu.cc | |
parent | ba98c6b6b8aa38140bd5acdbeb2a9ca419bc6188 (diff) |
Optimized the gradients of the sqrt, rsqrt, and inv functions
Change: 131463674
Diffstat (limited to 'tensorflow/core/kernels/cwise_op_gpu_sqrt.cu.cc')
-rw-r--r-- | tensorflow/core/kernels/cwise_op_gpu_sqrt.cu.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/cwise_op_gpu_sqrt.cu.cc b/tensorflow/core/kernels/cwise_op_gpu_sqrt.cu.cc index 8fba705343..dae93a0766 100644 --- a/tensorflow/core/kernels/cwise_op_gpu_sqrt.cu.cc +++ b/tensorflow/core/kernels/cwise_op_gpu_sqrt.cu.cc @@ -16,10 +16,12 @@ limitations under the License. #if GOOGLE_CUDA #include "tensorflow/core/kernels/cwise_ops_gpu_common.cu.h" +#include "tensorflow/core/kernels/cwise_ops_gpu_gradients.cu.h" namespace tensorflow { namespace functor { DEFINE_UNARY3(sqrt, Eigen::half, float, double); +DEFINE_SIMPLE_BINARY3(sqrt_grad, Eigen::half, float, double); } // namespace functor } // namespace tensorflow |