Searched defs:REGISTER_GPUCONCAT64 (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dconcat_lib_gpu_impl.cu.cc180 #define REGISTER_GPUCONCAT64(T) \ macro
208 TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPUCONCAT64);
209 TF_CALL_complex64(REGISTER_GPUCONCAT64);
210 TF_CALL_complex128(REGISTER_GPUCONCAT64);
211 TF_CALL_int64(REGISTER_GPUCONCAT64);
212 REGISTER_GPUCONCAT64(bfloat16);
213 REGISTER_GPUCONCAT64(bool);
230 #undef REGISTER_GPUCONCAT64 macro

Completed in 127 milliseconds