Home
last modified time | relevance | path

Searched refs:REGISTER_GPU_KERNEL (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dshape_ops.cc38 #define REGISTER_GPU_KERNEL(type) \ macro
52 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_GPU_KERNEL);
53 TF_CALL_bool(REGISTER_GPU_KERNEL);
54 TF_CALL_variant(REGISTER_GPU_KERNEL);
55 #undef REGISTER_GPU_KERNEL
127 #define REGISTER_GPU_KERNEL(type) \ macro
141 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_GPU_KERNEL);
142 TF_CALL_bool(REGISTER_GPU_KERNEL);
143 #undef REGISTER_GPU_KERNEL
205 #define REGISTER_GPU_KERNEL(type) \ macro
[all …]
Dcontrol_flow_ops.cc243 #define REGISTER_GPU_KERNEL(type) \ macro
257 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_GPU_KERNEL);
259 TF_CALL_QUANTIZED_TYPES(REGISTER_GPU_KERNEL);
261 REGISTER_GPU_KERNEL(bool);
263 TF_CALL_variant(REGISTER_GPU_KERNEL);
265 #undef REGISTER_GPU_KERNEL
307 #define REGISTER_GPU_KERNEL(type) \ macro
314 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_GPU_KERNEL);
316 REGISTER_GPU_KERNEL(bool);
318 TF_CALL_variant(REGISTER_GPU_KERNEL);
[all …]
Dsplit_lib_gpu.h48 #define REGISTER_GPU_KERNEL(T) \ macro
53 TF_CALL_bfloat16(REGISTER_GPU_KERNEL);
54 TF_CALL_uint8(REGISTER_GPU_KERNEL);
55 TF_CALL_GPU_ALL_TYPES(REGISTER_GPU_KERNEL);
56 #undef REGISTER_GPU_KERNEL
Dcwise_op_clip.cc188 #define REGISTER_GPU_KERNEL(type) \ macro
192 REGISTER_GPU_KERNEL(Eigen::half);
193 REGISTER_GPU_KERNEL(float);
194 REGISTER_GPU_KERNEL(double);
195 REGISTER_GPU_KERNEL(int8);
196 REGISTER_GPU_KERNEL(int16);
197 REGISTER_GPU_KERNEL(int64_t);
198 REGISTER_GPU_KERNEL(uint8);
199 REGISTER_GPU_KERNEL(uint16);
213 #undef REGISTER_GPU_KERNEL
Dl2loss_op_gpu.cu.cc63 #define REGISTER_GPU_KERNEL(T) \ macro
68 REGISTER_GPU_KERNEL(float);
69 REGISTER_GPU_KERNEL(double);
70 REGISTER_GPU_KERNEL(Eigen::half);
71 #undef REGISTER_GPU_KERNEL
Dsplit_lib_gpu.cu.cc244 #define REGISTER_GPU_KERNEL(T) template struct SplitOpGPULaunch<T>; macro
246 TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU_KERNEL);
247 TF_CALL_COMPLEX_TYPES(REGISTER_GPU_KERNEL);
248 TF_CALL_bfloat16(REGISTER_GPU_KERNEL);
249 #undef REGISTER_GPU_KERNEL
250 #define REGISTER_GPU_KERNEL(T) \ macro
254 TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU_KERNEL);
255 TF_CALL_COMPLEX_TYPES(REGISTER_GPU_KERNEL);
256 TF_CALL_bfloat16(REGISTER_GPU_KERNEL);
257 #undef REGISTER_GPU_KERNEL
Dreshape_op.cc32 #define REGISTER_GPU_KERNEL(type) \ macro
45 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_GPU_KERNEL);
46 TF_CALL_bool(REGISTER_GPU_KERNEL);
47 #undef REGISTER_GPU_KERNEL
Dbatch_norm_op.cc196 #define REGISTER_GPU_KERNEL(T) \ macro
202 TF_CALL_half(REGISTER_GPU_KERNEL);
203 TF_CALL_float(REGISTER_GPU_KERNEL);
204 #undef REGISTER_GPU_KERNEL
244 #define REGISTER_GPU_KERNEL(T) \ macro
250 TF_CALL_half(REGISTER_GPU_KERNEL);
251 TF_CALL_float(REGISTER_GPU_KERNEL);
252 #undef REGISTER_GPU_KERNEL
Didentity_op.cc56 #define REGISTER_GPU_KERNEL(type) \ macro
81 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_GPU_KERNEL);
82 REGISTER_GPU_KERNEL(Variant);
83 REGISTER_GPU_KERNEL(bool);
85 #undef REGISTER_GPU_KERNEL
Ddata_format_ops.cc304 #define REGISTER_GPU_KERNEL(T) \ macro
308 TF_CALL_int32(REGISTER_GPU_KERNEL);
309 TF_CALL_int64(REGISTER_GPU_KERNEL);
310 #undef REGISTER_GPU_KERNEL
312 #define REGISTER_GPU_KERNEL(T) \ macro
316 TF_CALL_int32(REGISTER_GPU_KERNEL);
317 TF_CALL_int64(REGISTER_GPU_KERNEL);
318 #undef REGISTER_GPU_KERNEL
Dcross_op.cc104 #define REGISTER_GPU_KERNEL(type) \ macro
109 TF_CALL_REAL_NUMBER_TYPES(REGISTER_GPU_KERNEL);
110 #undef REGISTER_GPU_KERNEL
Dsequence_ops.cc129 #define REGISTER_GPU_KERNEL(T) REGISTER_KERNEL(DEVICE_GPU, GPUDevice, T) macro
138 TF_CALL_float(REGISTER_GPU_KERNEL);
139 TF_CALL_double(REGISTER_GPU_KERNEL);
140 TF_CALL_int64(REGISTER_GPU_KERNEL);
156 #undef REGISTER_GPU_KERNEL
Dbias_op.cc286 #define REGISTER_GPU_KERNEL(type) \ macro
294 TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU_KERNEL);
295 REGISTER_GPU_KERNEL(int32);
296 #undef REGISTER_GPU_KERNEL
536 #define REGISTER_GPU_KERNEL(type) \ macro
541 TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU_KERNEL);
542 #undef REGISTER_GPU_KERNEL
Dpad_op.cc323 #define REGISTER_GPU_KERNEL(T) \ macro
351 TF_CALL_GPU_ALL_TYPES(REGISTER_GPU_KERNEL);
352 TF_CALL_int8(REGISTER_GPU_KERNEL);
353 TF_CALL_uint8(REGISTER_GPU_KERNEL);
Ddepthwise_conv_grad_op.cc727 #define REGISTER_GPU_KERNEL(T) \ macro
734 TF_CALL_half(REGISTER_GPU_KERNEL);
735 TF_CALL_float(REGISTER_GPU_KERNEL);
736 TF_CALL_double(REGISTER_GPU_KERNEL);
737 #undef REGISTER_GPU_KERNEL
1266 #define REGISTER_GPU_KERNEL(T) \ macro
1273 TF_CALL_half(REGISTER_GPU_KERNEL);
1274 TF_CALL_float(REGISTER_GPU_KERNEL);
1275 TF_CALL_double(REGISTER_GPU_KERNEL);
1276 #undef REGISTER_GPU_KERNEL
Ddepthwise_conv_op.cc525 #define REGISTER_GPU_KERNEL(T) \ macro
530 TF_CALL_half(REGISTER_GPU_KERNEL);
531 TF_CALL_float(REGISTER_GPU_KERNEL);
532 TF_CALL_double(REGISTER_GPU_KERNEL);
Dquantize_and_dequantize_op.cc468 #define REGISTER_GPU_KERNEL(T) \ macro
497 TF_CALL_float(REGISTER_GPU_KERNEL);
498 TF_CALL_double(REGISTER_GPU_KERNEL);
499 #undef REGISTER_GPU_KERNEL
/external/tensorflow/tensorflow/core/kernels/mlir_generated/
Dgpu_op_is_finite.cc24 REGISTER_GPU_KERNEL(IsFinite, DT_HALF, DT_BOOL);
26 REGISTER_GPU_KERNEL(IsFinite, DT_FLOAT, DT_BOOL);
28 REGISTER_GPU_KERNEL(IsFinite, DT_DOUBLE, DT_BOOL);
Dgpu_op_is_nan.cc24 REGISTER_GPU_KERNEL(IsNan, DT_HALF, DT_BOOL);
26 REGISTER_GPU_KERNEL(IsNan, DT_FLOAT, DT_BOOL);
28 REGISTER_GPU_KERNEL(IsNan, DT_DOUBLE, DT_BOOL);
Dgpu_op_is_inf.cc24 REGISTER_GPU_KERNEL(IsInf, DT_HALF, DT_BOOL);
26 REGISTER_GPU_KERNEL(IsInf, DT_FLOAT, DT_BOOL);
28 REGISTER_GPU_KERNEL(IsInf, DT_DOUBLE, DT_BOOL);
Dbase_gpu_op.h31 #define REGISTER_GPU_KERNEL(tf_op, input_type, output_type) \ macro
/external/tensorflow/tensorflow/core/kernels/image/
Dadjust_contrast_op.cc136 #define REGISTER_GPU_KERNEL(T) \ macro
140 REGISTER_GPU_KERNEL(uint8);
141 REGISTER_GPU_KERNEL(int8);
142 REGISTER_GPU_KERNEL(int16);
143 REGISTER_GPU_KERNEL(int32);
144 REGISTER_GPU_KERNEL(float);
145 REGISTER_GPU_KERNEL(double);
146 #undef REGISTER_GPU_KERNEL
Dmirror_pad_op.cc232 #define REGISTER_GPU_KERNEL(T) \ macro
246 TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU_KERNEL);
247 #undef REGISTER_GPU_KERNEL
439 #define REGISTER_GPU_KERNEL(T) \ macro
453 TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU_KERNEL);
454 #undef REGISTER_GPU_KERNEL
/external/tensorflow/tensorflow/core/kernels/rnn/
Dgru_ops.cc408 #define REGISTER_GPU_KERNEL(T) \ macro
413 REGISTER_GPU_KERNEL(float);
414 #undef REGISTER_GPU_KERNEL
441 #define REGISTER_GPU_KERNEL(T) \ macro
446 REGISTER_GPU_KERNEL(float);
447 #undef REGISTER_GPU_KERNEL
Dlstm_ops.cc506 #define REGISTER_GPU_KERNEL(T) \ macro
511 REGISTER_GPU_KERNEL(Eigen::half);
512 REGISTER_GPU_KERNEL(float);
513 #undef REGISTER_GPU_KERNEL
769 #define REGISTER_GPU_KERNEL(T) \ macro
774 REGISTER_GPU_KERNEL(Eigen::half);
775 REGISTER_GPU_KERNEL(float);
776 #undef REGISTER_GPU_KERNEL
1111 #define REGISTER_GPU_KERNEL(T) \ macro
1123 REGISTER_GPU_KERNEL(Eigen::half);
[all …]

12