Avoid using functions from the std namespace on GPU.

Change: 119695799

Avoid using functions from the std namespace on GPU.
Change: 119695799
c052df40 · Benoit Steiner · TensorFlower Gardener · 8217a41e · c052df40
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

tensorflow/core/kernels/sparse_tensor_dense_matmul_op.h tensorflow/core/kernels/sparse_tensor_dense_matmul_op.h +1 -1

未找到文件。
--- a/tensorflow/core/kernels/sparse_tensor_dense_matmul_op.h
+++ b/tensorflow/core/kernels/sparse_tensor_dense_matmul_op.h
@@ -66,7 +66,7 @@ EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE T MaybeConj(T v) {
 #define MAYBE_CONJ(T)                                         \
  template <>                                                 \
  EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE T MaybeConj<T>(T v) { \
-    return std::conj(v);                                      \
+    return Eigen::numext::conj(v);                            \
  }
 #endif