diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-21 18:46:35 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-21 18:50:27 -0700 |
| commit | a2fd40adcc714f18167acd9650e5442d4afd6a01 (patch) | |
| tree | 5c02772a56e0fdba2ef40573c5919a33e2edee7a /tensorflow/compiler/tests | |
| parent | 1cb8940078f6be9313899734e1307a69fffc4b6f (diff) | |
[tf:xla]Implement DivNoNan.
PiperOrigin-RevId: 214076068
Diffstat (limited to 'tensorflow/compiler/tests')
| -rw-r--r-- | tensorflow/compiler/tests/binary_ops_test.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/compiler/tests/binary_ops_test.py b/tensorflow/compiler/tests/binary_ops_test.py index 900e84ab58..e219cf3d88 100644 --- a/tensorflow/compiler/tests/binary_ops_test.py +++ b/tensorflow/compiler/tests/binary_ops_test.py @@ -560,6 +560,13 @@ class BinaryOpsTest(xla_test.XLATestCase): dtype(2), expected=np.array([[5], [2]], dtype=dtype)) + if dtype in [np.float32, np.float64]: + nums = np.arange(-10, 10, .25, dtype=dtype).reshape(80, 1) + divs = np.arange(-3, 3, .25, dtype=dtype).reshape(1, 24) + np_result = np.true_divide(nums, divs) + np_result[:, divs[0] == 0] = 0 + self._testBinary(gen_math_ops.div_no_nan, nums, divs, expected=np_result) + if dtype not in self.complex_types: # floordiv unsupported for complex. self._testBinary( gen_math_ops.floor_div, |