| /external/tensorflow/tensorflow/core/kernels/ |
| cwise_op_acos.cc | 19 REGISTER2(UnaryOp, CPU, "Acos", functor::acos, float, double);
22 REGISTER2(UnaryOp, GPU, "Acos", functor::acos, float, double);
26 REGISTER2(UnaryOp, SYCL, "Acos", functor::acos, float, double);
|
| cwise_op_asin.cc | 19 REGISTER2(UnaryOp, CPU, "Asin", functor::asin, float, double);
22 REGISTER2(UnaryOp, GPU, "Asin", functor::asin, float, double);
26 REGISTER2(UnaryOp, SYCL, "Asin", functor::asin, float, double);
|
| cwise_op_atan.cc | 19 REGISTER2(UnaryOp, CPU, "Atan", functor::atan, float, double);
22 REGISTER2(UnaryOp, GPU, "Atan", functor::atan, float, double);
26 REGISTER2(UnaryOp, SYCL, "Atan", functor::atan, float, double);
|
| cwise_op_tan.cc | 19 REGISTER2(UnaryOp, CPU, "Tan", functor::tan, float, double);
22 REGISTER2(UnaryOp, GPU, "Tan", functor::tan, float, double);
26 REGISTER2(UnaryOp, SYCL, "Tan", functor::tan, float, double);
|
| cwise_op_atan2.cc | 19 REGISTER2(BinaryOp, CPU, "Atan2", functor::atan2, float, double);
21 REGISTER2(BinaryOp, GPU, "Atan2", functor::atan2, float, double);
|
| cwise_op_igammas.cc | 19 REGISTER2(BinaryOp, CPU, "Igamma", functor::igamma, float, double);
20 REGISTER2(BinaryOp, CPU, "Igammac", functor::igammac, float, double);
|
| cwise_op_rint.cc | 19 REGISTER2(UnaryOp, CPU, "Rint", functor::rint, float, double);
21 REGISTER2(UnaryOp, GPU, "Rint", functor::rint, float, double);
|
| cwise_op_zeta.cc | 19 REGISTER2(BinaryOp, CPU, "Zeta", functor::zeta, float, double);
20 REGISTER2(BinaryOp, CPU, "Polygamma", functor::polygamma, float, double);
|
| cwise_op_acosh.cc | 24 REGISTER2(UnaryOp, SYCL, "Acosh", functor::acosh, float, double);
28 REGISTER2(UnaryOp, GPU, "Acosh", functor::acosh, float, double);
|
| cwise_op_asinh.cc | 24 REGISTER2(UnaryOp, SYCL, "Asinh", functor::asinh, float, double);
28 REGISTER2(UnaryOp, GPU, "Asinh", functor::asinh, float, double);
|
| cwise_op_atanh.cc | 24 REGISTER2(UnaryOp, SYCL, "Atanh", functor::atanh, float, double);
28 REGISTER2(UnaryOp, GPU, "Atanh", functor::atanh, float, double);
|
| cwise_op_mod.cc | 19 REGISTER2(BinaryOp, CPU, "Mod", functor::safe_mod, int32, int64);
20 REGISTER2(BinaryOp, CPU, "Mod", functor::fmod, float, double);
21 REGISTER2(BinaryOp, CPU, "TruncateMod", functor::safe_mod, int32, int64);
22 REGISTER2(BinaryOp, CPU, "TruncateMod", functor::fmod, float, double);
|
| cwise_op_pow.cc | 21 REGISTER2(BinaryOp, CPU, "Pow", functor::safe_pow, int32, int64);
28 REGISTER2(BinaryOp, SYCL, "Pow", functor::pow, float, double);
|
| cwise_op_floor_mod.cc | 19 REGISTER2(BinaryOp, CPU, "FloorMod", functor::safe_floor_mod, int32, int64);
20 REGISTER2(BinaryOp, CPU, "FloorMod", functor::floor_fmod, float, double);
|
| cwise_op_rsqrt.cc | 26 REGISTER2(UnaryOp, SYCL, "Rsqrt", functor::rsqrt, float, double);
36 REGISTER2(SimpleBinaryOp, SYCL, "RsqrtGrad", functor::rsqrt_grad, float,
|
| cwise_op_sqrt.cc | 27 REGISTER2(UnaryOp, SYCL, "Sqrt", functor::sqrt, float, double);
38 REGISTER2(SimpleBinaryOp, SYCL, "SqrtGrad", functor::sqrt_grad, float, double);
|
| cwise_op_ceil.cc | 26 REGISTER2(UnaryOp, SYCL, "Ceil", functor::ceil, float, double);
|
| cwise_op_cos.cc | 27 REGISTER2(UnaryOp, SYCL, "Cos", functor::cos, float, double);
|
| cwise_op_cosh.cc | 33 REGISTER2(UnaryOp, GPU, "Cosh", functor::cosh, float, double);
|
| cwise_op_exp.cc | 28 REGISTER2(UnaryOp, SYCL, "Exp", functor::exp, float, double);
|
| cwise_op_expm1.cc | 25 REGISTER2(UnaryOp, SYCL, "Expm1", functor::expm1, float, double);
|
| cwise_op_floor.cc | 25 REGISTER2(UnaryOp, SYCL, "Floor", functor::floor, float, double);
|
| cwise_op_isfinite.cc | 28 REGISTER2(UnaryOp, SYCL, "IsFinite", functor::isfinite, float, double);
|
| cwise_op_isinf.cc | 26 REGISTER2(UnaryOp, SYCL, "IsInf", functor::isinf, float, double);
|
| cwise_op_isnan.cc | 26 REGISTER2(UnaryOp, SYCL, "IsNan", functor::isnan, float, double);
|
