diff options
Diffstat (limited to 'tensorflow/core/kernels/cwise_op_cos.cc')
| -rw-r--r-- | tensorflow/core/kernels/cwise_op_cos.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/cwise_op_cos.cc b/tensorflow/core/kernels/cwise_op_cos.cc index 6958fa22b8..cd7dd976db 100644 --- a/tensorflow/core/kernels/cwise_op_cos.cc +++ b/tensorflow/core/kernels/cwise_op_cos.cc @@ -16,8 +16,8 @@ limitations under the License. #include "tensorflow/core/kernels/cwise_ops_common.h" namespace tensorflow { -REGISTER4(UnaryOp, CPU, "Cos", functor::cos, float, Eigen::half, double, - complex64); +REGISTER5(UnaryOp, CPU, "Cos", functor::cos, float, Eigen::half, double, + complex64, complex128); #if GOOGLE_CUDA REGISTER3(UnaryOp, GPU, "Cos", functor::cos, float, Eigen::half, double); #endif |