HomeSort by relevance Sort by last modified time
    Searched refs:rsqrt_grad (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
cwise_op_gpu_rsqrt.cu.cc 24 DEFINE_SIMPLE_BINARY3(rsqrt_grad, Eigen::half, float, double);
cwise_op_rsqrt.cc 29 REGISTER5(SimpleBinaryOp, CPU, "RsqrtGrad", functor::rsqrt_grad, float,
32 REGISTER3(SimpleBinaryOp, GPU, "RsqrtGrad", functor::rsqrt_grad, float,
36 REGISTER2(SimpleBinaryOp, SYCL, "RsqrtGrad", functor::rsqrt_grad, float,
cwise_ops_gradients.h 203 struct rsqrt_grad : base<T, Eigen::internal::scalar_rsqrt_gradient_op<T>> {}; struct in namespace:tensorflow::functor

Completed in 62 milliseconds