diff options
| author |  Robin Richtsfeld <robin.richtsfeld@gmail.com> | 2018-05-26 01:38:33 +0200 |
|---|---|---|
| committer |  Rasmus Munk Larsen <rmlarsen@google.com> | 2018-05-25 16:38:33 -0700 |
| commit | 83116bafebb500fa963809599b7f2583367c92d6 (patch) | |
| tree | 1fdaff5404a07c83daccaae9c88f68851fce2a36 /tensorflow/core/kernels/scatter_nd_op_gpu.cu.cc | |
| parent | 38926b8a0fa89bef74085be0e321c13e739795d4 (diff) | |
Fix of issue #13164 (Merges #13382) (#16368)
* tf.gather int64 GPU, tf.gather_nd int32/int64 GPU, tf.scatter_nd int32 GPU
* Fix tf.gather test
Diffstat (limited to 'tensorflow/core/kernels/scatter_nd_op_gpu.cu.cc')
| -rw-r--r-- | tensorflow/core/kernels/scatter_nd_op_gpu.cu.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/scatter_nd_op_gpu.cu.cc b/tensorflow/core/kernels/scatter_nd_op_gpu.cu.cc index a3c21edc15..08b657f4c3 100644 --- a/tensorflow/core/kernels/scatter_nd_op_gpu.cu.cc +++ b/tensorflow/core/kernels/scatter_nd_op_gpu.cu.cc @@ -170,6 +170,7 @@ struct ScatterNdFunctor<GPUDevice, T, Index, op, IXDIM> { DECLARE_GPU_SPECS_INDEX(T, int32); \ DECLARE_GPU_SPECS_INDEX(T, int64) +TF_CALL_int32(DECLARE_GPU_SPECS); TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPECS); TF_CALL_complex64(DECLARE_GPU_SPECS); TF_CALL_complex128(DECLARE_GPU_SPECS); |