Home
last modified time | relevance | path

Searched refs:REGISTER_GPU_HOST_KERNEL (Results 1 – 3 of 3) 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.cc97 #define REGISTER_GPU_HOST_KERNEL(type) \ macro
123 REGISTER_GPU_HOST_KERNEL(int32);
124 REGISTER_GPU_HOST_KERNEL(tstring);
125 REGISTER_GPU_HOST_KERNEL(ResourceHandle);
127 #undef REGISTER_GPU_HOST_KERNEL
Dshape_ops.cc339 #define REGISTER_GPU_HOST_KERNEL(type) \ macro
347 REGISTER_GPU_HOST_KERNEL(int32);
348 REGISTER_GPU_HOST_KERNEL(bool);
349 REGISTER_GPU_HOST_KERNEL(tstring);
350 REGISTER_GPU_HOST_KERNEL(ResourceHandle);
352 #undef REGISTER_GPU_HOST_KERNEL