diff options
Diffstat (limited to 'tensorflow/core/kernels/cwise_ops_gpu_common.cu.h')
| -rw-r--r-- | tensorflow/core/kernels/cwise_ops_gpu_common.cu.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/cwise_ops_gpu_common.cu.h b/tensorflow/core/kernels/cwise_ops_gpu_common.cu.h index d91d0faa86..6b23fb5785 100644 --- a/tensorflow/core/kernels/cwise_ops_gpu_common.cu.h +++ b/tensorflow/core/kernels/cwise_ops_gpu_common.cu.h @@ -34,6 +34,7 @@ namespace functor { typedef Eigen::GpuDevice GPUDevice; typedef std::complex<float> complex64; +typedef std::complex<double> complex128; // Partial specialization of UnaryFunctor<Device=GPUDevice, Functor>. template <typename Functor> @@ -149,6 +150,9 @@ struct BinaryFunctor<GPUDevice, Functor, NDIMS, has_errors> { #define DEFINE_BINARY9(F, T0, T1, T2, T3, T4, T5, T6, T7, T8) \ DEFINE_BINARY4(F, T0, T1, T2, T3); \ DEFINE_BINARY5(F, T4, T5, T6, T7, T8) +#define DEFINE_BINARY10(F, T0, T1, T2, T3, T4, T5, T6, T7, T8, T9) \ + DEFINE_BINARY5(F, T0, T1, T2, T3, T4); \ + DEFINE_BINARY5(F, T5, T6, T7, T8, T9) } // end namespace functor } // end namespace tensorflow |