| /external/tensorflow/tensorflow/core/kernels/ |
| cwise_op_gpu_inverse.cu.cc | 24 DEFINE_SIMPLE_BINARY3(inverse_grad, Eigen::half, float, double);
|
| cwise_op_reciprocal.cc | 26 REGISTER5(SimpleBinaryOp, CPU, "InvGrad", functor::inverse_grad, float,
29 REGISTER3(SimpleBinaryOp, GPU, "InvGrad", functor::inverse_grad, float,
43 REGISTER5(SimpleBinaryOp, CPU, "ReciprocalGrad", functor::inverse_grad, float,
46 REGISTER3(SimpleBinaryOp, GPU, "ReciprocalGrad", functor::inverse_grad, float,
50 REGISTER(SimpleBinaryOp, SYCL, "ReciprocalGrad", functor::inverse_grad, float);
|
| cwise_ops_gradients.h | 196 struct inverse_grad : base<T, Eigen::internal::scalar_inverse_gradient_op<T>> { struct in namespace:tensorflow::functor
|
