diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-01-29 17:25:45 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-29 17:29:31 -0800 |
commit | aca9f4257f064d381be171a7ff2ee114002a8fab (patch) | |
tree | 10b94f143de5982c7391cea067e59e1ed9aa2222 /tensorflow/compiler/tests/BUILD | |
parent | 18447358543598abc3296f5df9e7774528dc53f0 (diff) |
[TF:XLA] Complete the TriangularSolve implementation.
The previous version only handled the case of left_side=false, lower=true, transpose_a=true, conjugate_a=false. This updated implementation handles all 16 combinations of those boolean options, and also instantiates the corresponding MatrixTriangularSolve TF op.
To improve compile times and potentially FLOP performance, when lower=true the within-block subroutine used on the diagonal blocks is now a left-looking variant implemented with an XLA HLO While loop.
This update also slightly generalizes BatchDot in tf2xla/lib to accept separate conjugation arguments (in addition to transpose arguments).
PiperOrigin-RevId: 183756639
Diffstat (limited to 'tensorflow/compiler/tests/BUILD')
-rw-r--r-- | tensorflow/compiler/tests/BUILD | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/tensorflow/compiler/tests/BUILD b/tensorflow/compiler/tests/BUILD index a3a82df9ad..9e64f3e9a3 100644 --- a/tensorflow/compiler/tests/BUILD +++ b/tensorflow/compiler/tests/BUILD @@ -145,6 +145,21 @@ tf_xla_py_test( ) tf_xla_py_test( + name = "matrix_triangular_solve_op_test", + size = "small", + srcs = ["matrix_triangular_solve_op_test.py"], + tags = ["optonly"], + deps = [ + ":xla_test", + "//tensorflow/python:array_ops", + "//tensorflow/python:framework_for_generated_wrappers", + "//tensorflow/python:math_ops", + "//tensorflow/python:platform_test", + "//tensorflow/python:training", + ], +) + +tf_xla_py_test( name = "clustering_test", size = "small", srcs = ["clustering_test.py"], |