experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Enforce xla in xla_control_flow_ops_test.

Browse Source 
I actually don't understand how one can avoid using xla in an xla
test. But the test seems to pass now.

PiperOrigin-RevId: 324379840
Change-Id: I62c8638183148a95c9c5ab930fa64cfb4c66926c
This commit is contained in:
Yunxing Dai 2020-08-01 02:40:22 -07:00 committed by TensorFlower Gardener
parent 8f00058182
commit 44597e39fb
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/python/ops/parallel_for/xla_control_flow_ops_test.py
View File

@ -233,8 +233,7 @@ class WhileV2Test(PForTestCase):
body,
[True, 0, 0.])
# b/155430349: Enabling forrce_xla=True triggers a CHECK in debug mode.
self._test_loop_fn(loop_fn, 3, force_xla=False)
self._test_loop_fn(loop_fn, 3, force_xla=True)
if __name__ == "__main__":