Fix of issue #13164 (Merges #13382) (#16368)
authorRobin Richtsfeld <robin.richtsfeld@gmail.com>
Fri, 25 May 2018 23:38:33 +0000 (01:38 +0200)
committerRasmus Munk Larsen <rmlarsen@google.com>
Fri, 25 May 2018 23:38:33 +0000 (16:38 -0700)
commit83116bafebb500fa963809599b7f2583367c92d6
tree1fdaff5404a07c83daccaae9c88f68851fce2a36
parent38926b8a0fa89bef74085be0e321c13e739795d4
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
12 files changed:
tensorflow/core/kernels/dense_update_functor_gpu.cu.cc
tensorflow/core/kernels/gather_functor.cc
tensorflow/core/kernels/gather_functor_gpu.cu.cc
tensorflow/core/kernels/gather_nd_op.cc
tensorflow/core/kernels/gather_nd_op_gpu.cu.cc
tensorflow/core/kernels/gather_op.cc
tensorflow/core/kernels/scatter_nd_op.cc
tensorflow/core/kernels/scatter_nd_op_gpu.cu.cc
tensorflow/python/kernel_tests/gather_nd_op_test.py
tensorflow/python/kernel_tests/gather_op_test.py
tensorflow/python/kernel_tests/scatter_nd_ops_test.py
tensorflow/python/kernel_tests/scatter_ops_test.py