diff options
Diffstat (limited to 'tensorflow/core/kernels/cwise_op_floor.cc')
| -rw-r--r-- | tensorflow/core/kernels/cwise_op_floor.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/cwise_op_floor.cc b/tensorflow/core/kernels/cwise_op_floor.cc index 7205399888..8d4ae1b22a 100644 --- a/tensorflow/core/kernels/cwise_op_floor.cc +++ b/tensorflow/core/kernels/cwise_op_floor.cc @@ -16,8 +16,8 @@ limitations under the License. #include "tensorflow/core/kernels/cwise_ops_common.h" namespace tensorflow { -REGISTER3(UnaryOp, CPU, "Floor", functor::floor, Eigen::half, float, double); +REGISTER3(UnaryOp, CPU, "Floor", functor::floor, float, Eigen::half, double); #if GOOGLE_CUDA -REGISTER3(UnaryOp, GPU, "Floor", functor::floor, Eigen::half, float, double); +REGISTER3(UnaryOp, GPU, "Floor", functor::floor, float, Eigen::half, double); #endif } // namespace tensorflow |