experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Internal change

Browse Source 
PiperOrigin-RevId: 324726407
Change-Id: Ie720508eeb1375dfd82d7ac5ef208c5f2e6edb45
This commit is contained in:
A. Unique TensorFlower 2020-08-03 18:29:31 -07:00 committed by TensorFlower Gardener
parent d353f49989
commit e4592dad25
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/kernels/cwise_op_exp.cc
View File

@ -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,

1
tensorflow/python/kernel_tests/cwise_ops_unary_test.py
View File

@ -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)