diff options
author | 2018-04-04 11:35:05 -0700 | |
---|---|---|
committer | 2018-04-04 11:37:52 -0700 | |
commit | 73b0540f81d6a5b88557b69df1bc3da25ca81a1b (patch) | |
tree | 22d6f768c5645eb1d7415e8593f095de63133c57 | |
parent | 0fef384fb94f83abd9c787b3c8ab7abc1f7ade95 (diff) |
Also registers the gpu resourcegather kernel for half (no idea why it was missing)
PiperOrigin-RevId: 191621111
-rw-r--r-- | tensorflow/core/kernels/resource_variable_ops.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/resource_variable_ops.cc b/tensorflow/core/kernels/resource_variable_ops.cc index 5c54609ee6..f49a05c70a 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 |