Searched refs:REGISTER_CPU (Results 1 – 16 of 16) sorted by relevance
81 #define REGISTER_CPU(T) \ macro85 TF_CALL_half(REGISTER_CPU);86 TF_CALL_float(REGISTER_CPU);87 TF_CALL_double(REGISTER_CPU);89 #undef REGISTER_CPU90 #define REGISTER_CPU(T) \ macro94 TF_CALL_half(REGISTER_CPU);95 TF_CALL_float(REGISTER_CPU);96 TF_CALL_double(REGISTER_CPU);
565 #define REGISTER_CPU(T) \ macro589 TF_CALL_half(REGISTER_CPU);590 TF_CALL_bfloat16(REGISTER_CPU);591 TF_CALL_int32(REGISTER_CPU);592 TF_CALL_int64(REGISTER_CPU);605 TF_CALL_complex64(REGISTER_CPU);606 TF_CALL_complex128(REGISTER_CPU);607 TF_CALL_double(REGISTER_CPU);615 TF_CALL_float(REGISTER_CPU);616 TF_CALL_double(REGISTER_CPU);[all …]
191 #define REGISTER_CPU(TYPE) REGISTER(CPU, TYPE) macro196 TF_CALL_half(REGISTER_CPU);197 TF_CALL_bfloat16(REGISTER_CPU);198 TF_CALL_float(REGISTER_CPU);199 TF_CALL_double(REGISTER_CPU);215 #undef REGISTER_CPU
198 #define REGISTER_CPU(TYPE) \ macro211 TF_CALL_half(REGISTER_CPU);212 TF_CALL_bfloat16(REGISTER_CPU);213 TF_CALL_float(REGISTER_CPU);214 TF_CALL_double(REGISTER_CPU);225 #undef REGISTER_CPU
136 #define REGISTER_CPU(T) \ macro143 TF_CALL_NUMBER_TYPES(REGISTER_CPU);144 TF_CALL_QUANTIZED_TYPES(REGISTER_CPU);145 #undef REGISTER_CPU
223 #define REGISTER_CPU(T) \ macro231 TF_CALL_float(REGISTER_CPU);234 TF_CALL_double(REGISTER_CPU);235 TF_CALL_complex64(REGISTER_CPU);236 TF_CALL_complex128(REGISTER_CPU);
128 #define REGISTER_CPU(T) \ macro133 TF_CALL_half(REGISTER_CPU);134 TF_CALL_float(REGISTER_CPU);135 TF_CALL_double(REGISTER_CPU);
286 #define REGISTER_CPU(type) REGISTER_KERNEL(type, CPU) macro287 TF_CALL_POD_STRING_TYPES(REGISTER_CPU);288 REGISTER_CPU(Variant);289 #undef REGISTER_CPU342 #define REGISTER_CPU(type) REGISTER_KERNEL(type, CPU) macro343 TF_CALL_POD_TYPES(REGISTER_CPU);344 #undef REGISTER_CPU
286 #define REGISTER_CPU(T) \ macro290 TF_CALL_float(REGISTER_CPU);291 TF_CALL_half(REGISTER_CPU);293 #undef REGISTER_CPU509 #define REGISTER_CPU(T) \ macro513 TF_CALL_float(REGISTER_CPU);514 TF_CALL_half(REGISTER_CPU);516 #undef REGISTER_CPU
110 #define REGISTER_CPU(T) \ macro119 TF_CALL_float(REGISTER_CPU);120 TF_CALL_double(REGISTER_CPU);
298 #define REGISTER_CPU(T) \ macro308 TF_CALL_float(REGISTER_CPU);309 #undef REGISTER_CPU
421 #define REGISTER_CPU(T) \ macro426 REGISTER_CPU(float);427 REGISTER_CPU(Eigen::half);428 REGISTER_CPU(double);430 #undef REGISTER_CPU
158 #define REGISTER_CPU(TypeT, TypeIndex) \ macro168 REGISTER_CPU(T, int64); \169 REGISTER_CPU(T, int32)
310 #define REGISTER_CPU(type) \ macro336 TF_CALL_ALL_TYPES(REGISTER_CPU);337 #undef REGISTER_CPU
563 #define REGISTER_CPU(T) \ macro573 TF_CALL_half(REGISTER_CPU);574 TF_CALL_float(REGISTER_CPU);
533 #define REGISTER_CPU(T) \ macro541 TF_CALL_half(REGISTER_CPU);542 TF_CALL_float(REGISTER_CPU);543 TF_CALL_double(REGISTER_CPU);