From 9616883b82ee81b4b03faafb219d5aa23c275ca8 Mon Sep 17 00:00:00 2001 From: Yong Tang Date: Thu, 18 Jun 2020 16:17:52 +0000 Subject: [PATCH] Add complex64 and complex128 gpu support for tensor_scatter_nd_add This PR adds complex64 and complex128 gpu support for tensor_scatter_nd_add, as was raised in 40577. This PR fixes 40577. Signed-off-by: Yong Tang --- tensorflow/core/kernels/inplace_ops_functor_gpu.cu.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tensorflow/core/kernels/inplace_ops_functor_gpu.cu.cc b/tensorflow/core/kernels/inplace_ops_functor_gpu.cu.cc index bfffadecdf1..8bee7dbee67 100644 --- a/tensorflow/core/kernels/inplace_ops_functor_gpu.cu.cc +++ b/tensorflow/core/kernels/inplace_ops_functor_gpu.cu.cc @@ -197,6 +197,8 @@ Status DoCopy(const Device& d, const Tensor& x, Tensor* y) { CASE(float) CASE(double) CASE(Eigen::half) + CASE(complex64) + CASE(complex128) CASE(int64) #undef CASE default: