Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dcontrol_flow_ops.cc250 #define REGISTER_GPU_REF_KERNEL(type) \ macro
258 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_GPU_REF_KERNEL);
260 TF_CALL_QUANTIZED_TYPES(REGISTER_GPU_REF_KERNEL);
262 REGISTER_GPU_REF_KERNEL(bool);
266 #undef REGISTER_GPU_REF_KERNEL
310 #define REGISTER_GPU_REF_KERNEL(type) \ macro
315 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_GPU_REF_KERNEL);
317 REGISTER_GPU_REF_KERNEL(bool);
321 #undef REGISTER_GPU_REF_KERNEL
367 #define REGISTER_GPU_REF_KERNEL(type) \ macro
[all …]