Home
last modified time | relevance | path

Searched refs:REGISTER_CPU_KERNEL (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcwise_op_clip.cc169 #define REGISTER_CPU_KERNEL(type) \ macro
174 REGISTER_CPU_KERNEL(Eigen::half);
175 REGISTER_CPU_KERNEL(float);
176 REGISTER_CPU_KERNEL(double);
177 REGISTER_CPU_KERNEL(bfloat16);
178 REGISTER_CPU_KERNEL(int8);
179 REGISTER_CPU_KERNEL(int16);
180 REGISTER_CPU_KERNEL(int32);
181 REGISTER_CPU_KERNEL(int64);
182 REGISTER_CPU_KERNEL(uint8);
[all …]
Dsequence_ops.cc100 #define REGISTER_CPU_KERNEL(T) REGISTER_KERNEL(DEVICE_CPU, T) macro
103 TF_CALL_float(REGISTER_CPU_KERNEL);
104 TF_CALL_double(REGISTER_CPU_KERNEL);
105 TF_CALL_int32(REGISTER_CPU_KERNEL);
106 TF_CALL_int64(REGISTER_CPU_KERNEL);
118 #undef REGISTER_CPU_KERNEL
173 #define REGISTER_CPU_KERNEL(T) REGISTER_KERNEL_ALL_NUMS(DEVICE_CPU, T) macro
174 TF_CALL_float(REGISTER_CPU_KERNEL);
175 TF_CALL_double(REGISTER_CPU_KERNEL);
185 #undef REGISTER_CPU_KERNEL
Dquantized_reshape_op.cc46 #define REGISTER_CPU_KERNEL(type) \ macro
53 REGISTER_CPU_KERNEL(::tensorflow::quint8);
54 REGISTER_CPU_KERNEL(::tensorflow::qint32);
56 #undef REGISTER_CPU_KERNEL
Dragged_range_op.cc117 #define REGISTER_CPU_KERNEL(TYPE) \ macro
128 TF_CALL_float(REGISTER_CPU_KERNEL);
129 TF_CALL_double(REGISTER_CPU_KERNEL);
130 TF_CALL_int32(REGISTER_CPU_KERNEL);
131 TF_CALL_int64(REGISTER_CPU_KERNEL);
132 #undef REGISTER_CPU_KERNEL
Dunsorted_segment_join_op.cc152 #define REGISTER_CPU_KERNEL(indices_type, num_segments_type) \ macro
160 REGISTER_CPU_KERNEL(int32, int32);
161 REGISTER_CPU_KERNEL(int32, int64);
162 REGISTER_CPU_KERNEL(int64, int32);
163 REGISTER_CPU_KERNEL(int64, int64);
164 #undef REGISTER_CPU_KERNEL
Dragged_gather_op.cc289 #define REGISTER_CPU_KERNEL(value_type) \ macro
294 TF_CALL_POD_TYPES(REGISTER_CPU_KERNEL);
295 TF_CALL_tstring(REGISTER_CPU_KERNEL);
296 TF_CALL_QUANTIZED_TYPES(REGISTER_CPU_KERNEL);
297 TF_CALL_quint16(REGISTER_CPU_KERNEL);
298 TF_CALL_qint16(REGISTER_CPU_KERNEL);
299 #undef REGISTER_CPU_KERNEL
Dcross_op.cc83 #define REGISTER_CPU_KERNEL(type) \ macro
87 TF_CALL_REAL_NUMBER_TYPES(REGISTER_CPU_KERNEL);
88 #undef REGISTER_CPU_KERNEL
Disotonic_regression_op.cc207 #define REGISTER_CPU_KERNEL(Tin, Tout) \ macro
215 #define REGISTER_CPU_SAME_KERNEL(T) REGISTER_CPU_KERNEL(T, T)
219 #define REGISTER_CPU_KERNEL_FLOAT(Tin) REGISTER_CPU_KERNEL(Tin, float)
224 #define REGISTER_CPU_KERNEL_DOUBLE(Tin) REGISTER_CPU_KERNEL(Tin, double)
Dragged_tensor_to_tensor_op.cc555 #define REGISTER_CPU_KERNEL(value_type) \ macro
559 TF_CALL_POD_TYPES(REGISTER_CPU_KERNEL);
560 TF_CALL_string(REGISTER_CPU_KERNEL);
561 TF_CALL_QUANTIZED_TYPES(REGISTER_CPU_KERNEL);
562 TF_CALL_quint16(REGISTER_CPU_KERNEL);
563 TF_CALL_qint16(REGISTER_CPU_KERNEL);
565 #undef REGISTER_CPU_KERNEL
Dcheck_numerics_op.cc363 #define REGISTER_CPU_KERNEL(T) \ macro
367 TF_CALL_half(REGISTER_CPU_KERNEL);
368 TF_CALL_bfloat16(REGISTER_CPU_KERNEL);
369 TF_CALL_float(REGISTER_CPU_KERNEL);
370 TF_CALL_double(REGISTER_CPU_KERNEL);
Dedit_distance_op.cc238 #define REGISTER_CPU_KERNEL(T) \ macro
243 TF_CALL_POD_STRING_TYPES(REGISTER_CPU_KERNEL);
245 #undef REGISTER_CPU_KERNEL
Ddepthwise_conv_grad_op.cc708 #define REGISTER_CPU_KERNEL(T) \ macro
714 TF_CALL_half(REGISTER_CPU_KERNEL);
715 TF_CALL_float(REGISTER_CPU_KERNEL);
717 TF_CALL_double(REGISTER_CPU_KERNEL);
719 #undef REGISTER_CPU_KERNEL
1239 #define REGISTER_CPU_KERNEL(T) \ macro
1245 TF_CALL_half(REGISTER_CPU_KERNEL);
1246 TF_CALL_float(REGISTER_CPU_KERNEL);
1248 TF_CALL_double(REGISTER_CPU_KERNEL);
1250 #undef REGISTER_CPU_KERNEL
Ddepthwise_conv_op.cc492 #define REGISTER_CPU_KERNEL(T) \ macro
497 TF_CALL_half(REGISTER_CPU_KERNEL);
498 TF_CALL_float(REGISTER_CPU_KERNEL);
500 TF_CALL_double(REGISTER_CPU_KERNEL);
Dconstant_op.cc182 #define REGISTER_CPU_KERNEL(TYPE) REGISTER_KERNEL(CPU, TYPE) macro
183 TF_CALL_ALL_TYPES(REGISTER_CPU_KERNEL);
191 #undef REGISTER_CPU_KERNEL
Dquantize_and_dequantize_op.cc410 #define REGISTER_CPU_KERNEL(T) \ macro
430 TF_CALL_float(REGISTER_CPU_KERNEL);
431 TF_CALL_double(REGISTER_CPU_KERNEL);
432 #undef REGISTER_CPU_KERNEL
Davgpooling_op.cc397 #define REGISTER_CPU_KERNEL(T) \ macro
404 TF_CALL_float(REGISTER_CPU_KERNEL);
405 TF_CALL_double(REGISTER_CPU_KERNEL);
406 TF_CALL_half(REGISTER_CPU_KERNEL);
Dconv_ops_3d.cc190 #define REGISTER_CPU_KERNEL(T) \ macro
194 TF_CALL_half(REGISTER_CPU_KERNEL);
195 TF_CALL_float(REGISTER_CPU_KERNEL);
196 TF_CALL_double(REGISTER_CPU_KERNEL);
197 #undef REGISTER_CPU_KERNEL
Dconv_grad_ops_3d.cc596 #define REGISTER_CPU_KERNEL(T) \ macro
624 TF_CALL_half(REGISTER_CPU_KERNEL);
625 TF_CALL_float(REGISTER_CPU_KERNEL);
626 TF_CALL_double(REGISTER_CPU_KERNEL);
627 #undef REGISTER_CPU_KERNEL
1016 #define REGISTER_CPU_KERNEL(T) \ macro
1045 TF_CALL_float(REGISTER_CPU_KERNEL);
1046 TF_CALL_double(REGISTER_CPU_KERNEL);
1047 #undef REGISTER_CPU_KERNEL
1051 #define REGISTER_CPU_KERNEL(T) \ macro
[all …]
Dquantized_resize_bilinear_op.cc736 #define REGISTER_CPU_KERNEL(type) \ macro
743 REGISTER_CPU_KERNEL(::tensorflow::quint8);
744 REGISTER_CPU_KERNEL(::tensorflow::qint32);
745 REGISTER_CPU_KERNEL(float);
/external/tensorflow/tensorflow/core/kernels/image/
Ddraw_bounding_box_op.cc204 #define REGISTER_CPU_KERNEL(T) \ macro
211 TF_CALL_half(REGISTER_CPU_KERNEL);
212 TF_CALL_float(REGISTER_CPU_KERNEL);
/external/tensorflow/tensorflow/core/kernels/neon/
Dneon_depthwise_conv_op.cc195 #define REGISTER_CPU_KERNEL(T) \ macro
202 TF_CALL_float(REGISTER_CPU_KERNEL);
/external/tensorflow/tensorflow/core/kernels/mlir_generated/
Dbase_cpu_op.h35 #define REGISTER_CPU_KERNEL(tf_op, input_type, output_type) \ macro