diff options
| author |  Tom Hennigan <tomhennigan@google.com> | 2018-06-05 08:42:28 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-05 08:45:12 -0700 |
| commit | e1f31d40b9d12e687100a689bc5439d78702124c (patch) | |
| tree | 2d7a6580e39f7b7f13654599df433ba2c6ac645d /tensorflow/python/framework/test_util.py | |
| parent | 3653e80488f490ad744410a92ac287acf7035bda (diff) | |
Expose `@tfe.run_all_tests_in_graph_and_eager_modes`.
PiperOrigin-RevId: 199302255
Diffstat (limited to 'tensorflow/python/framework/test_util.py')
| -rw-r--r-- | tensorflow/python/framework/test_util.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/framework/test_util.py b/tensorflow/python/framework/test_util.py index b56483f373..0c06d9aa41 100644 --- a/tensorflow/python/framework/test_util.py +++ b/tensorflow/python/framework/test_util.py @@ -644,6 +644,7 @@ def assert_no_garbage_created(f): def run_all_in_graph_and_eager_modes(cls): + """Execute all test methods in the given class with and without eager.""" base_decorator = run_in_graph_and_eager_modes() for name, value in cls.__dict__.copy().items(): if callable(value) and name.startswith("test"): |