diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-12-21 14:50:09 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-12-21 14:54:21 -0800 |
commit | e21952f86328d16d1a8a4ac5e100917a251d093f (patch) | |
tree | 1d9274230d915810cd78563f9952ee7ecf006408 /tensorflow/core/kernels/scatter_nd_op_gpu.cu.cc | |
parent | 7989a3aa026f93c1f45e1baf4ac37ad2a79156f9 (diff) |
Automated g4 rollback of changelist 179782067
PiperOrigin-RevId: 179861781
Diffstat (limited to 'tensorflow/core/kernels/scatter_nd_op_gpu.cu.cc')
-rw-r--r-- | tensorflow/core/kernels/scatter_nd_op_gpu.cu.cc | 21 |
1 files changed, 0 insertions, 21 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..31f74671ca 100644 --- a/tensorflow/core/kernels/scatter_nd_op_gpu.cu.cc +++ b/tensorflow/core/kernels/scatter_nd_op_gpu.cu.cc @@ -55,27 +55,6 @@ struct LeftUpdate<T, scatter_nd_op::UpdateOp::SUB> { } }; -// Specializations for std::complex, updating real and imaginary part -// individually. Even though this is not an atomic op anymore, it is safe -// because there is only one type of op per kernel. -template <typename T> -struct LeftUpdate<std::complex<T>, scatter_nd_op::UpdateOp::ADD> { - EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC void operator()( - std::complex<T>* out, const std::complex<T>& val) { - T* ptr = reinterpret_cast<T*>(out); - CudaAtomicAdd(ptr, val.real()); - CudaAtomicAdd(ptr, val.imag()); - } -}; - -template <typename T> -struct LeftUpdate<std::complex<T>, scatter_nd_op::UpdateOp::SUB> { - EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC void operator()( - std::complex<T>* out, const std::complex<T>& val) { - LeftUpdate<std::complex<T>, scatter_nd_op::UpdateOp::ADD>()(out, -val); - } -}; - } // namespace template <typename T, typename Index, scatter_nd_op::UpdateOp op, int IXDIM> |