aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/kernels/scatter_nd_op_gpu.cu.cc
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2017-12-21 14:50:09 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-12-21 14:54:21 -0800
commite21952f86328d16d1a8a4ac5e100917a251d093f (patch)
tree1d9274230d915810cd78563f9952ee7ecf006408 /tensorflow/core/kernels/scatter_nd_op_gpu.cu.cc
parent7989a3aa026f93c1f45e1baf4ac37ad2a79156f9 (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.cc21
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>