diff options
author | RJ Ryan <rryan@alum.mit.edu> | 2016-03-22 13:15:33 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-03-22 15:24:15 -0700 |
commit | fab3bd9b2bc3bb9f44ecc4e96b97081d4dd4d8d2 (patch) | |
tree | 0cb4e542cd5fdf9a94f92d0adbd278c4273a0c49 | |
parent | 4bc8285bb65da03862f4131bf7b4df6fe32de560 (diff) |
Test tf.complex_abs in cwise_ops_test.
Change: 117862980
-rw-r--r-- | tensorflow/python/kernel_tests/cwise_ops_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/kernel_tests/cwise_ops_test.py b/tensorflow/python/kernel_tests/cwise_ops_test.py index 888e651eea..1b15e47558 100644 --- a/tensorflow/python/kernel_tests/cwise_ops_test.py +++ b/tensorflow/python/kernel_tests/cwise_ops_test.py @@ -62,7 +62,7 @@ class UnaryOpTest(tf.test.TestCase): return # Return early if x.dtype == np.complex64 and tf_func in ( - tf.sign, tf.abs, _ABS, tf.sqrt, tf.rsqrt, tf.log): + tf.sign, tf.sqrt, tf.rsqrt, tf.log): return # Return early if x.dtype == np.float32 or x.dtype == np.complex64: @@ -218,7 +218,7 @@ class UnaryOpTest(tf.test.TestCase): x = np.complex(1, 1) * np.arange(-3, 3).reshape(1, 3, 2).astype( np.complex64) y = x + 0.5 # no zeros - self._compareCpu(x, np.abs, tf.abs) + self._compareCpu(x, np.abs, tf.complex_abs) self._compareCpu(x, np.abs, _ABS) self._compareCpu(x, np.negative, tf.neg) self._compareCpu(x, np.negative, _NEG) |