diff --git a/tensorflow/core/kernels/cwise_op_rsqrt.cc b/tensorflow/core/kernels/cwise_op_rsqrt.cc index e051e4d702a..21e3bf4d33f 100644 --- a/tensorflow/core/kernels/cwise_op_rsqrt.cc +++ b/tensorflow/core/kernels/cwise_op_rsqrt.cc @@ -16,15 +16,15 @@ limitations under the License. #include "tensorflow/core/kernels/cwise_ops_common.h" namespace tensorflow { -REGISTER6(UnaryOp, CPU, "Rsqrt", functor::rsqrt, float, Eigen::half, bfloat16, - double, complex64, complex128); +REGISTER5(UnaryOp, CPU, "Rsqrt", functor::rsqrt, float, Eigen::half, double, + complex64, complex128); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM REGISTER3(UnaryOp, GPU, "Rsqrt", functor::rsqrt, float, Eigen::half, double); #endif -REGISTER6(SimpleBinaryOp, CPU, "RsqrtGrad", functor::rsqrt_grad, float, - Eigen::half, bfloat16, double, complex64, complex128); +REGISTER5(SimpleBinaryOp, CPU, "RsqrtGrad", functor::rsqrt_grad, float, + Eigen::half, double, complex64, complex128); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM REGISTER3(SimpleBinaryOp, GPU, "RsqrtGrad", functor::rsqrt_grad, float, Eigen::half, double); diff --git a/tensorflow/python/kernel_tests/cwise_ops_unary_test.py b/tensorflow/python/kernel_tests/cwise_ops_unary_test.py index c0f61f0aa60..9d46ed35639 100644 --- a/tensorflow/python/kernel_tests/cwise_ops_unary_test.py +++ b/tensorflow/python/kernel_tests/cwise_ops_unary_test.py @@ -405,7 +405,6 @@ class UnaryOpTest(test.TestCase): self._compareCpu(z, compute_f32(np.log), math_ops.log) self._compareCpu(z, compute_f32(np.log1p), math_ops.log1p) self._compareCpu(y, np.sign, math_ops.sign) - self._compareCpu(z, self._rsqrt, math_ops.rsqrt) self._compareBoth(x, compute_f32(np.sin), math_ops.sin) self._compareBoth(x, compute_f32(np.cos), math_ops.cos) self._compareBoth(x, compute_f32(np.tan), math_ops.tan)