experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Also registers the gpu resourcegather kernel for half (no idea why it was missing)

Browse Source 
PiperOrigin-RevId: 191621111
This commit is contained in:
Alexandre Passos 2018-04-04 11:35:05 -07:00 committed by TensorFlower Gardener
parent 0fef384fb9
commit 73b0540f81
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/kernels/resource_variable_ops.cc
View File

@ -605,7 +605,7 @@ TF_CALL_QUANTIZED_TYPES(REGISTER_GATHER_CPU);
#if GOOGLE_CUDA #if GOOGLE_CUDA
#define REGISTER_GATHER_GPU(type) REGISTER_GATHER_ALL_INDICES(GPU, type) #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 #endif // GOOGLE_CUDA