diff options
Diffstat (limited to 'tensorflow/core/kernels/concat_lib_gpu_impl.cu.cc')
| -rw-r--r-- | tensorflow/core/kernels/concat_lib_gpu_impl.cu.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/concat_lib_gpu_impl.cu.cc b/tensorflow/core/kernels/concat_lib_gpu_impl.cu.cc index f971637d5d..0f7adaf24a 100644 --- a/tensorflow/core/kernels/concat_lib_gpu_impl.cu.cc +++ b/tensorflow/core/kernels/concat_lib_gpu_impl.cu.cc @@ -203,24 +203,28 @@ TF_CALL_complex64(REGISTER_GPUCONCAT32); TF_CALL_complex128(REGISTER_GPUCONCAT32); TF_CALL_int64(REGISTER_GPUCONCAT32); REGISTER_GPUCONCAT32(bfloat16); +REGISTER_GPUCONCAT32(bool); TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPUCONCAT64); TF_CALL_complex64(REGISTER_GPUCONCAT64); TF_CALL_complex128(REGISTER_GPUCONCAT64); TF_CALL_int64(REGISTER_GPUCONCAT64); REGISTER_GPUCONCAT64(bfloat16); +REGISTER_GPUCONCAT64(bool); TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU32); TF_CALL_complex64(REGISTER_GPU32); TF_CALL_complex128(REGISTER_GPU32); TF_CALL_int64(REGISTER_GPU32); REGISTER_GPU32(bfloat16); +REGISTER_GPU32(bool); TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU64); TF_CALL_complex64(REGISTER_GPU64); TF_CALL_complex128(REGISTER_GPU64); TF_CALL_int64(REGISTER_GPU64); REGISTER_GPU64(bfloat16); +REGISTER_GPU64(bool); #undef REGISTER_GPUCONCAT32 #undef REGISTER_GPUCONCAT64 |