Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dcontrol_flow_ops.cc123 #define REGISTER_GPU_HOST_KERNEL(type) \ macro
150 REGISTER_GPU_HOST_KERNEL(int32);
152 REGISTER_GPU_HOST_KERNEL(tstring);
154 REGISTER_GPU_HOST_KERNEL(ResourceHandle);
156 #undef REGISTER_GPU_HOST_KERNEL
272 #define REGISTER_GPU_HOST_KERNEL(type) \ macro
288 REGISTER_GPU_HOST_KERNEL(int32);
289 REGISTER_GPU_HOST_KERNEL(tstring);
290 REGISTER_GPU_HOST_KERNEL(ResourceHandle);
292 #undef REGISTER_GPU_HOST_KERNEL
[all …]
Didentity_op.cc124 #define REGISTER_GPU_HOST_KERNEL(type) \ macro
156 REGISTER_GPU_HOST_KERNEL(int32);
157 REGISTER_GPU_HOST_KERNEL(tstring);
158 REGISTER_GPU_HOST_KERNEL(ResourceHandle);
160 #undef REGISTER_GPU_HOST_KERNEL