/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
|