diff options
author | 2016-05-23 11:39:39 -0800 | |
---|---|---|
committer | 2016-05-23 12:42:36 -0700 | |
commit | 892ca4ddc12852a7b4633fd08f163941356cb4e6 (patch) | |
tree | be913f46bb9323685c5a807a89fca6dc52a25504 /tensorflow/core/kernels/cwise_op_square.cc | |
parent | 76d90938f95a14a5723c253ec8529e93939a25e2 (diff) |
Merge changes from github.
Change: 123026122
Diffstat (limited to 'tensorflow/core/kernels/cwise_op_square.cc')
-rw-r--r-- | tensorflow/core/kernels/cwise_op_square.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/cwise_op_square.cc b/tensorflow/core/kernels/cwise_op_square.cc index 80c8423cf9..f03d2f11d4 100644 --- a/tensorflow/core/kernels/cwise_op_square.cc +++ b/tensorflow/core/kernels/cwise_op_square.cc @@ -16,8 +16,8 @@ limitations under the License. #include "tensorflow/core/kernels/cwise_ops_common.h" namespace tensorflow { -REGISTER6(UnaryOp, CPU, "Square", functor::square, float, Eigen::half, double, - int32, complex64, int64); +REGISTER7(UnaryOp, CPU, "Square", functor::square, float, Eigen::half, double, + int32, complex64, complex128, int64); #if GOOGLE_CUDA REGISTER4(UnaryOp, GPU, "Square", functor::square, float, Eigen::half, double, int64); |