diff options
author | Peter Hawkins <phawkins@google.com> | 2018-07-10 17:44:41 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-10 19:38:32 -0700 |
commit | 6f207c867450a68e6d6e4179a85622741c93d710 (patch) | |
tree | f6daf2544f0bff482c0079bba4fff7204f8c890a /tensorflow/compiler/tests/BUILD | |
parent | 4e78d0d41fb8d5d40d088d3ba2cdc531059733e8 (diff) |
[TF:XLA] Add implementation of block Householder QR decomposition.
PiperOrigin-RevId: 204044417
Diffstat (limited to 'tensorflow/compiler/tests/BUILD')
-rw-r--r-- | tensorflow/compiler/tests/BUILD | 23 |
1 files changed, 22 insertions, 1 deletions
diff --git a/tensorflow/compiler/tests/BUILD b/tensorflow/compiler/tests/BUILD index 8b25147899..e8e19f055e 100644 --- a/tensorflow/compiler/tests/BUILD +++ b/tensorflow/compiler/tests/BUILD @@ -235,7 +235,7 @@ tf_xla_py_test( tf_xla_py_test( name = "cholesky_op_test", - size = "small", + size = "medium", srcs = ["cholesky_op_test.py"], tags = ["optonly"], deps = [ @@ -665,6 +665,27 @@ tf_xla_py_test( ) tf_xla_py_test( + name = "qr_op_test", + size = "medium", + srcs = ["qr_op_test.py"], + disabled_backends = [ + # Test is very slow on CPU. + "cpu", + "cpu_ondemand", + ], + 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", + "@absl_py//absl/testing:parameterized", + ], +) + +tf_xla_py_test( name = "random_ops_test", size = "small", srcs = ["random_ops_test.py"], |