diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-08-01 16:02:16 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-01 16:09:25 -0700 |
commit | a4f755678102a4fb8861fec3238ddc78bb9b2a6e (patch) | |
tree | 72bb65c44d0bda91a304f548991d2c9e6a31910b /tensorflow/python/kernel_tests/linalg_grad_test.py | |
parent | bed00207976a70370cb6e3615c7ad56a5547cf45 (diff) |
Created pure TensorFlow op for MatrixExponential, allowing gradients and GPU ops.
PiperOrigin-RevId: 207010324
Diffstat (limited to 'tensorflow/python/kernel_tests/linalg_grad_test.py')
-rw-r--r-- | tensorflow/python/kernel_tests/linalg_grad_test.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/python/kernel_tests/linalg_grad_test.py b/tensorflow/python/kernel_tests/linalg_grad_test.py index 6f401358a2..0e4e58409e 100644 --- a/tensorflow/python/kernel_tests/linalg_grad_test.py +++ b/tensorflow/python/kernel_tests/linalg_grad_test.py @@ -26,6 +26,7 @@ from tensorflow.python.ops import gradient_checker from tensorflow.python.ops import gradients_impl from tensorflow.python.ops import linalg_ops from tensorflow.python.ops import math_ops +from tensorflow.python.ops.linalg import linalg_impl from tensorflow.python.platform import test as test_lib @@ -173,6 +174,10 @@ if __name__ == '__main__': _AddTest(MatrixUnaryFunctorGradientTest, 'MatrixInverseGradient', name, _GetMatrixUnaryFunctorGradientTest(linalg_ops.matrix_inverse, dtype, shape)) + _AddTest(MatrixUnaryFunctorGradientTest, 'MatrixExponentialGradient', + name, + _GetMatrixUnaryFunctorGradientTest( + linalg_impl.matrix_exponential, dtype, shape)) _AddTest( MatrixUnaryFunctorGradientTest, 'MatrixDeterminantGradient', name, _GetMatrixUnaryFunctorGradientTest(linalg_ops.matrix_determinant, |