/external/tensorflow/tensorflow/core/kernels/ |
cwise_op_acosh.cc | 20 REGISTER4(UnaryOp, CPU, "Acosh", functor::acosh, float, double, complex64,
|
cwise_op_asinh.cc | 20 REGISTER4(UnaryOp, CPU, "Asinh", functor::asinh, float, double, complex64,
|
cwise_op_atanh.cc | 20 REGISTER4(UnaryOp, CPU, "Atanh", functor::atanh, float, double, complex64,
|
cwise_op_not_equal_to_1.cc | 22 REGISTER4(BinaryOp, GPU, "NotEqual", functor::not_equal_to, float, Eigen::half,
|
cwise_op_add_2.cc | 32 REGISTER4(BinaryOp, GPU, "Add", functor::add, uint8, int64, complex64, 34 REGISTER4(BinaryOp, GPU, "AddV2", functor::add, uint8, int64, complex64,
|
cwise_op_cosh.cc | 19 REGISTER4(UnaryOp, CPU, "Cosh", functor::cosh, float, double, complex64,
|
cwise_op_pow.cc | 24 REGISTER4(BinaryOp, GPU, "Pow", functor::pow, float, Eigen::half, double,
|
cwise_op_sinh.cc | 19 REGISTER4(UnaryOp, CPU, "Sinh", functor::sinh, float, double, complex64,
|
cwise_op_maximum.cc | 22 REGISTER4(BinaryOp, GPU, "Maximum", functor::maximum, float, Eigen::half,
|
cwise_op_minimum.cc | 22 REGISTER4(BinaryOp, GPU, "Minimum", functor::minimum, float, Eigen::half,
|
cwise_op_square.cc | 23 REGISTER4(UnaryOp, GPU, "Square", functor::square, float, Eigen::half, double,
|
cwise_op_squared_difference.cc | 22 REGISTER4(BinaryOp, GPU, "SquaredDifference", functor::squared_difference,
|
cwise_op_reciprocal.cc | 22 REGISTER4(UnaryOp, GPU, "Inv", functor::inverse, float, Eigen::half, double, 36 REGISTER4(UnaryOp, GPU, "Reciprocal", functor::inverse, float, Eigen::half,
|
cwise_op_abs.cc | 26 REGISTER4(UnaryOp, GPU, "Abs", functor::abs, float, Eigen::half, double, int64);
|
cwise_op_floor_div.cc | 25 REGISTER4(BinaryOp, GPU, "FloorDiv", functor::floor_div, uint8, uint16, int16,
|
cwise_op_mul_1.cc | 30 REGISTER4(BinaryOp, GPU, "Mul", functor::mul, float, Eigen::half, double,
|
cwise_op_sub.cc | 23 REGISTER4(BinaryOp, CPU, "Sub", functor::sub, int8, uint8, int16, uint16);
|
cwise_op_div.cc | 30 REGISTER4(BinaryOp, GPU, "TruncateDiv", functor::div, uint8, uint16, int16,
|
cwise_op_equal_to_1.cc | 28 REGISTER4(BinaryOp, GPU, "Equal", functor::equal_to, float, Eigen::half, double,
|
cwise_ops_common.h | 561 #define REGISTER4(OP, D, N, F, T0, T1, T2, T3) REGISTER(OP, D, N, F, T0) 577 #define REGISTER4(OP, D, N, F, T0, T1, T2, T3) \ 587 REGISTER4(OP, D, N, F, T0, T1, T2, T3) \ 590 REGISTER4(OP, D, N, F, T0, T1, T2, T3) \ 591 REGISTER4(OP, D, N, F, T4, T5, T6, T7) 594 REGISTER4(OP, D, N, F, T5, T6, T7, T8)
|