diff options
Diffstat (limited to 'tensorflow/python/kernel_tests/cwise_ops_test.py')
| -rw-r--r-- | tensorflow/python/kernel_tests/cwise_ops_test.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/python/kernel_tests/cwise_ops_test.py b/tensorflow/python/kernel_tests/cwise_ops_test.py index c32149e880..c097177849 100644 --- a/tensorflow/python/kernel_tests/cwise_ops_test.py +++ b/tensorflow/python/kernel_tests/cwise_ops_test.py @@ -779,8 +779,9 @@ class BinaryOpTest(tf.test.TestCase): x = (1 + np.linspace(0, 5, np.prod(xs))).astype(dtype).reshape(xs) y = (1 + np.linspace(0, 5, np.prod(ys))).astype(dtype).reshape(ys) self._compareCpu(x, y, np_func, tf_func) - if x.dtype in (np.float16, np.float32, np.float64, np.complex64, - np.complex128): + if x.dtype in (np.float16, np.float32, np.float64): + # TODO(aselle): Make the test work for dtypes: + # (np.complex64, np.complex128). if tf_func not in (_FLOORDIV, tf.floordiv): if x.dtype == np.float16: # Compare fp16 theoretical gradients to fp32 numerical gradients, |