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

  /external/tensorflow/tensorflow/core/kernels/
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_op_sigmoid.cc 30 REGISTER5(SimpleBinaryOp, CPU, "SigmoidGrad", functor::sigmoid_grad, float,
33 REGISTER3(SimpleBinaryOp, GPU, "SigmoidGrad", functor::sigmoid_grad, float,
37 REGISTER(SimpleBinaryOp, SYCL, "SigmoidGrad", functor::sigmoid_grad, float);
cwise_op_sqrt.cc 30 REGISTER5(SimpleBinaryOp, CPU, "SqrtGrad", functor::sqrt_grad, float,
33 REGISTER3(SimpleBinaryOp, GPU, "SqrtGrad", functor::sqrt_grad, float,
38 REGISTER2(SimpleBinaryOp, SYCL, "SqrtGrad", functor::sqrt_grad, float, double);
cwise_op_tanh.cc 31 REGISTER5(SimpleBinaryOp, CPU, "TanhGrad", functor::tanh_grad, float,
34 REGISTER3(SimpleBinaryOp, GPU, "TanhGrad", functor::tanh_grad, float,
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_common.h 195 class SimpleBinaryOp : public OpKernel {
200 explicit SimpleBinaryOp(OpKernelConstruction* ctx) : OpKernel(ctx) {}

Completed in 199 milliseconds