diff options
Diffstat (limited to 'tensorflow/compiler/tests/binary_ops_test.py')
-rw-r--r-- | tensorflow/compiler/tests/binary_ops_test.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow/compiler/tests/binary_ops_test.py b/tensorflow/compiler/tests/binary_ops_test.py index 83cfd2ea75..c23ee5f037 100644 --- a/tensorflow/compiler/tests/binary_ops_test.py +++ b/tensorflow/compiler/tests/binary_ops_test.py @@ -114,6 +114,14 @@ class BinaryOpsTest(XLATestCase): expected=np.array([0.4, 1.2, 2.4, 4, 5, 6], dtype=dtype)) self._testBinary( + gen_nn_ops._selu_grad, + np.array([1, 2, 3, 4, 5, 6], dtype=dtype), + np.array([-.6, -.4, -.2, .2, .4, .6], dtype=dtype), + expected=np.array( + [1.158099340847, 2.7161986816948, 4.67429802254, + 4.202803949422, 5.2535049367774, 6.30420592413], dtype=dtype)) + + self._testBinary( gen_nn_ops._relu_grad, np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10], dtype=dtype), np.array([0, 0, 0, 0, 0, 0.1, 0.3, 0.5, 0.7, 0.9], dtype=dtype), |