From f2f582b3c00744c5e8857a309d38b374a5bd60fe Mon Sep 17 00:00:00 2001 From: Benoit Steiner Date: Fri, 26 Aug 2016 17:21:44 -0800 Subject: Optimized the gradients of the sqrt, rsqrt, and inv functions Change: 131463674 --- tensorflow/core/kernels/cwise_op_gpu_sqrt.cu.cc | 2 ++ 1 file changed, 2 insertions(+) (limited to 'tensorflow/core/kernels/cwise_op_gpu_sqrt.cu.cc') 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 -- cgit v1.2.3