From e4592dad255abb37e6ad675d4caa50161aba7e82 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Mon, 3 Aug 2020 18:29:31 -0700 Subject: [PATCH] Internal change PiperOrigin-RevId: 324726407 Change-Id: Ie720508eeb1375dfd82d7ac5ef208c5f2e6edb45 --- tensorflow/core/kernels/cwise_op_exp.cc | 4 ++-- tensorflow/python/kernel_tests/cwise_ops_unary_test.py | 1 - 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/tensorflow/core/kernels/cwise_op_exp.cc b/tensorflow/core/kernels/cwise_op_exp.cc index 48b6823cbdc..2b157f0e7a9 100644 --- a/tensorflow/core/kernels/cwise_op_exp.cc +++ b/tensorflow/core/kernels/cwise_op_exp.cc @@ -16,8 +16,8 @@ limitations under the License. #include "tensorflow/core/kernels/cwise_ops_common.h" namespace tensorflow { -REGISTER6(UnaryOp, CPU, "Exp", functor::exp, float, Eigen::half, double, - bfloat16, complex64, complex128); +REGISTER5(UnaryOp, CPU, "Exp", functor::exp, float, Eigen::half, double, + complex64, complex128); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM REGISTER5(UnaryOp, GPU, "Exp", functor::exp, 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 368f3509dc6..df848a653d4 100644 --- a/tensorflow/python/kernel_tests/cwise_ops_unary_test.py +++ b/tensorflow/python/kernel_tests/cwise_ops_unary_test.py @@ -389,7 +389,6 @@ class UnaryOpTest(test.TestCase): 2).reshape(1, 3, 2).astype(dtypes_lib.bfloat16.as_numpy_dtype) self._compareCpu(x, np.abs, math_ops.abs) self._compareCpu(x, np.abs, _ABS) - self._compareCpu(x, np.exp, math_ops.exp) self._compareBoth(x, np.negative, math_ops.negative) self._compareBoth(x, np.negative, _NEG)