From 73b0540f81d6a5b88557b69df1bc3da25ca81a1b Mon Sep 17 00:00:00 2001 From: Alexandre Passos Date: Wed, 4 Apr 2018 11:35:05 -0700 Subject: [PATCH] Also registers the gpu resourcegather kernel for half (no idea why it was missing) PiperOrigin-RevId: 191621111 --- tensorflow/core/kernels/resource_variable_ops.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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