Home
last modified time | relevance | path

Searched refs:REGISTER_CPU (Results 1 – 25 of 59) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/risc/experimental/
Drisc_dot_op.cc34 #define REGISTER_CPU(T) \ macro
39 REGISTER_CPU(bfloat16);
40 REGISTER_CPU(Eigen::half);
41 REGISTER_CPU(float);
42 REGISTER_CPU(double);
Drisc_add_op.cc36 #define REGISTER_CPU(T) \ macro
41 REGISTER_CPU(bfloat16);
42 REGISTER_CPU(Eigen::half);
43 REGISTER_CPU(float);
44 REGISTER_CPU(double);
Drisc_binary_comparison_op.cc36 #define REGISTER_CPU(T) \ macro
41 REGISTER_CPU(bfloat16);
42 REGISTER_CPU(Eigen::half);
43 REGISTER_CPU(float);
44 REGISTER_CPU(double);
Drisc_cholesky_op.cc36 #define REGISTER_CPU(T) \ macro
41 REGISTER_CPU(bfloat16);
42 REGISTER_CPU(Eigen::half);
43 REGISTER_CPU(float);
44 REGISTER_CPU(double);
Drisc_slice_op.cc34 #define REGISTER_CPU(type) \ macro
39 REGISTER_CPU(bfloat16);
40 REGISTER_CPU(Eigen::half);
41 REGISTER_CPU(float);
42 REGISTER_CPU(double);
Drisc_ceil_op.cc36 #define REGISTER_CPU(T) \ macro
41 REGISTER_CPU(bfloat16);
42 REGISTER_CPU(Eigen::half);
43 REGISTER_CPU(float);
44 REGISTER_CPU(double);
Drisc_sign_op.cc36 #define REGISTER_CPU(T) \ macro
41 REGISTER_CPU(bfloat16);
42 REGISTER_CPU(Eigen::half);
43 REGISTER_CPU(float);
44 REGISTER_CPU(double);
Drisc_concat_op.cc34 #define REGISTER_CPU(T) \ macro
39 REGISTER_CPU(bfloat16);
40 REGISTER_CPU(Eigen::half);
41 REGISTER_CPU(float);
42 REGISTER_CPU(double);
Drisc_neg_op.cc36 #define REGISTER_CPU(T) \ macro
41 REGISTER_CPU(bfloat16);
42 REGISTER_CPU(Eigen::half);
43 REGISTER_CPU(float);
44 REGISTER_CPU(double);
Drisc_max_op.cc36 #define REGISTER_CPU(T) \ macro
41 REGISTER_CPU(bfloat16);
42 REGISTER_CPU(Eigen::half);
43 REGISTER_CPU(float);
44 REGISTER_CPU(double);
Drisc_div_op.cc36 #define REGISTER_CPU(T) \ macro
41 REGISTER_CPU(bfloat16);
42 REGISTER_CPU(Eigen::half);
43 REGISTER_CPU(float);
44 REGISTER_CPU(double);
Drisc_rem_op.cc36 #define REGISTER_CPU(T) \ macro
41 REGISTER_CPU(bfloat16);
42 REGISTER_CPU(Eigen::half);
43 REGISTER_CPU(float);
44 REGISTER_CPU(double);
Drisc_abs_op.cc36 #define REGISTER_CPU(T) \ macro
41 REGISTER_CPU(bfloat16);
42 REGISTER_CPU(Eigen::half);
43 REGISTER_CPU(float);
44 REGISTER_CPU(double);
Drisc_floor_op.cc36 #define REGISTER_CPU(T) \ macro
41 REGISTER_CPU(bfloat16);
42 REGISTER_CPU(Eigen::half);
43 REGISTER_CPU(float);
44 REGISTER_CPU(double);
Drisc_log_op.cc36 #define REGISTER_CPU(T) \ macro
41 REGISTER_CPU(bfloat16);
42 REGISTER_CPU(Eigen::half);
43 REGISTER_CPU(float);
44 REGISTER_CPU(double);
Drisc_cos_op.cc36 #define REGISTER_CPU(T) \ macro
41 REGISTER_CPU(bfloat16);
42 REGISTER_CPU(Eigen::half);
43 REGISTER_CPU(float);
44 REGISTER_CPU(double);
Drisc_binary_arithmetic_op.cc36 #define REGISTER_CPU(T) \ macro
41 REGISTER_CPU(bfloat16);
42 REGISTER_CPU(Eigen::half);
43 REGISTER_CPU(float);
44 REGISTER_CPU(double);
Drisc_sub_op.cc36 #define REGISTER_CPU(T) \ macro
41 REGISTER_CPU(bfloat16);
42 REGISTER_CPU(Eigen::half);
43 REGISTER_CPU(float);
44 REGISTER_CPU(double);
Drisc_min_op.cc36 #define REGISTER_CPU(T) \ macro
41 REGISTER_CPU(bfloat16);
42 REGISTER_CPU(Eigen::half);
43 REGISTER_CPU(float);
44 REGISTER_CPU(double);
Drisc_broadcast_op.cc34 #define REGISTER_CPU(T) \ macro
39 REGISTER_CPU(bfloat16);
40 REGISTER_CPU(Eigen::half);
41 REGISTER_CPU(float);
42 REGISTER_CPU(double);
Drisc_mul_op.cc36 #define REGISTER_CPU(T) \ macro
41 REGISTER_CPU(bfloat16);
42 REGISTER_CPU(Eigen::half);
43 REGISTER_CPU(float);
44 REGISTER_CPU(double);
Drisc_pow_op.cc36 #define REGISTER_CPU(T) \ macro
41 REGISTER_CPU(bfloat16);
42 REGISTER_CPU(Eigen::half);
43 REGISTER_CPU(float);
44 REGISTER_CPU(double);
Drisc_pool_op.cc34 #define REGISTER_CPU(T) \ macro
39 REGISTER_CPU(bfloat16);
40 REGISTER_CPU(Eigen::half);
41 REGISTER_CPU(float);
42 REGISTER_CPU(double);
Drisc_exp_op.cc36 #define REGISTER_CPU(T) \ macro
41 REGISTER_CPU(bfloat16);
42 REGISTER_CPU(Eigen::half);
43 REGISTER_CPU(float);
44 REGISTER_CPU(double);
Drisc_pad_op.cc34 #define REGISTER_CPU(T) \ macro
41 REGISTER_CPU(bfloat16);
42 REGISTER_CPU(Eigen::half);
43 REGISTER_CPU(float);
44 REGISTER_CPU(double);

123