diff options
Diffstat (limited to 'tensorflow/compiler/tests/unary_ops_test.py')
-rw-r--r-- | tensorflow/compiler/tests/unary_ops_test.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/compiler/tests/unary_ops_test.py b/tensorflow/compiler/tests/unary_ops_test.py index ce35eb9197..81ff18f302 100644 --- a/tensorflow/compiler/tests/unary_ops_test.py +++ b/tensorflow/compiler/tests/unary_ops_test.py @@ -230,6 +230,11 @@ class UnaryOpsTest(XLATestCase): expected=np.array([[-0.63212056, 0, 1]], dtype=dtype)) self._assertOpOutputMatchesExpected( + nn_ops.selu, + np.array([[-1, 0, 1]], dtype=dtype), + expected=np.array([[-1.11133074, 0., 1.05070099]], dtype=dtype)) + + self._assertOpOutputMatchesExpected( nn_ops.relu, np.array([[-1, 1]], dtype=dtype), expected=np.array([[0, 1]], dtype=dtype)) |