diff --git a/tensorflow/core/kernels/resource_variable_ops.cc b/tensorflow/core/kernels/resource_variable_ops.cc index 5c54609ee67..f49a05c70ad 100644 --- a/tensorflow/core/kernels/resource_variable_ops.cc +++ b/tensorflow/core/kernels/resource_variable_ops.cc @@ -605,7 +605,7 @@ TF_CALL_QUANTIZED_TYPES(REGISTER_GATHER_CPU); #if GOOGLE_CUDA #define REGISTER_GATHER_GPU(type) REGISTER_GATHER_ALL_INDICES(GPU, type) -TF_CALL_GPU_NUMBER_TYPES_NO_HALF(REGISTER_GATHER_GPU); +TF_CALL_GPU_NUMBER_TYPES(REGISTER_GATHER_GPU); #endif // GOOGLE_CUDA