| /external/tensorflow/tensorflow/core/kernels/ |
| cwise_op_gpu_isinf.cu.cc | 22 DEFINE_UNARY3(isinf, Eigen::half, float, double);
|
| cwise_op_gpu_isnan.cu.cc | 22 DEFINE_UNARY3(isnan, Eigen::half, float, double);
|
| cwise_op_gpu_lgamma.cu.cc | 22 DEFINE_UNARY3(lgamma, Eigen::half, float, double);
|
| cwise_op_gpu_log.cu.cc | 22 DEFINE_UNARY3(log, Eigen::half, float, double);
|
| cwise_op_gpu_log1p.cu.cc | 22 DEFINE_UNARY3(log1p, Eigen::half, float, double);
|
| cwise_op_gpu_maximum.cu.cc | 22 DEFINE_BINARY4(maximum, Eigen::half, float, double, int64);
|
| cwise_op_gpu_minimum.cu.cc | 22 DEFINE_BINARY4(minimum, Eigen::half, float, double, int64);
|
| cwise_op_gpu_pow.cu.cc | 22 DEFINE_BINARY4(pow, Eigen::half, float, double, int64);
|
| cwise_op_gpu_sin.cu.cc | 22 DEFINE_UNARY3(sin, Eigen::half, float, double);
|
| cwise_op_gpu_square.cu.cc | 22 DEFINE_UNARY4(square, Eigen::half, float, double, int64);
|
| cwise_op_gpu_squared_difference.cu.cc | 22 DEFINE_BINARY4(squared_difference, float, Eigen::half, double, int64);
|
| cwise_op_gpu_tan.cu.cc | 22 DEFINE_UNARY3(tan, Eigen::half, float, double);
|
| scan_ops_gpu_half.cu.cc | 26 GpuDevice, Eigen::internal::SumReducer<Eigen::half>, Eigen::half>;
28 GpuDevice, Eigen::internal::ProdReducer<Eigen::half>, Eigen::half>;
|
| cwise_op_digamma.cc | 19 REGISTER3(UnaryOp, CPU, "Digamma", functor::digamma, float, Eigen::half,
22 REGISTER3(UnaryOp, GPU, "Digamma", functor::digamma, float, Eigen::half,
|
| cwise_op_erf.cc | 19 REGISTER3(UnaryOp, CPU, "Erf", functor::erf, float, Eigen::half, double);
21 REGISTER3(UnaryOp, GPU, "Erf", functor::erf, float, Eigen::half, double);
|
| cwise_op_erfc.cc | 19 REGISTER3(UnaryOp, CPU, "Erfc", functor::erfc, float, Eigen::half, double);
21 REGISTER3(UnaryOp, GPU, "Erfc", functor::erfc, float, Eigen::half, double);
|
| cwise_op_lgamma.cc | 19 REGISTER3(UnaryOp, CPU, "Lgamma", functor::lgamma, float, Eigen::half, double);
21 REGISTER3(UnaryOp, GPU, "Lgamma", functor::lgamma, float, Eigen::half, double);
|
| tile_functor_gpu_half.cu.cc | 26 template struct Tile<GpuDevice, Eigen::half, int32>;
27 template struct Tile<GpuDevice, Eigen::half, int64>;
|
| cwise_op_gpu_abs.cu.cc | 22 DEFINE_UNARY6(abs, Eigen::half, float, double, int64, complex64, complex128);
|
| cwise_op_gpu_exp.cu.cc | 22 DEFINE_UNARY5(exp, Eigen::half, float, double, complex64, complex128);
|
| cwise_op_gpu_round.cu.cc | 22 DEFINE_UNARY5(round, Eigen::half, float, double, int32, int64);
|
| cwise_op_gpu_sign.cu.cc | 22 DEFINE_UNARY6(sign, Eigen::half, float, double, int64, complex64, complex128);
|
| cwise_op_gpu_sub.cu.cc | 22 DEFINE_BINARY6(sub, Eigen::half, float, double, int64, complex64, complex128);
|
| cwise_op_gpu_xdivy.cu.cc | 22 DEFINE_BINARY5(xdivy, Eigen::half, float, double, complex64, complex128);
|
| cwise_op_gpu_xlogy.cu.cc | 22 DEFINE_BINARY5(xlogy, Eigen::half, float, double, complex64, complex128);
|
