Searched refs:REGISTER_GPUCONCAT32 (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dconcat_lib_gpu_impl.cu.cc173 #define REGISTER_GPUCONCAT32(T) \ macro
201 TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPUCONCAT32);
202 TF_CALL_complex64(REGISTER_GPUCONCAT32);
203 TF_CALL_complex128(REGISTER_GPUCONCAT32);
204 TF_CALL_int64(REGISTER_GPUCONCAT32);
205 REGISTER_GPUCONCAT32(bfloat16);
206 REGISTER_GPUCONCAT32(bool);
229 #undef REGISTER_GPUCONCAT32 macro

Completed in 264 milliseconds