diff options
author | 2018-03-21 04:09:32 -0700 | |
---|---|---|
committer | 2018-03-21 04:12:21 -0700 | |
commit | e69000c347ddf023a3b1926d812881fd8c5a055b (patch) | |
tree | 13108c33f78294891fa33d3e16332eace2e84326 /tensorflow/compiler/tests/BUILD | |
parent | d5d74b0aaaa221ad64aa1d86cb8428df2b885cf7 (diff) |
We were ValueOrDie()ing in one place, and TF_CHECK_OK()ing in another. Both should gracefully return an error condition.
Add some tests to check this.
PiperOrigin-RevId: 189888700
Diffstat (limited to 'tensorflow/compiler/tests/BUILD')
-rw-r--r-- | tensorflow/compiler/tests/BUILD | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/tensorflow/compiler/tests/BUILD b/tensorflow/compiler/tests/BUILD index bbb6089ea8..26d4ca0c13 100644 --- a/tensorflow/compiler/tests/BUILD +++ b/tensorflow/compiler/tests/BUILD @@ -192,6 +192,26 @@ tf_xla_py_test( ) tf_xla_py_test( + name = "oom_test", + size = "medium", + srcs = ["oom_test.py"], + disabled_backends = [ + "cpu", + "cpu_ondemand", + ], + deps = [ + ":xla_test", + "//tensorflow/python:array_ops", + "//tensorflow/python:array_ops_gen", + "//tensorflow/python:framework_for_generated_wrappers", + "//tensorflow/python:gradient_checker", + "//tensorflow/python:gradients", + "//tensorflow/python:math_ops", + "//tensorflow/python:platform_test", + ], +) + +tf_xla_py_test( name = "conv2d_test", size = "medium", srcs = ["conv2d_test.py"], |