Searched refs:force_xla (Results 1 – 1 of 1) sorted by relevance
159 def _test_loop_fn(self, loop_fn, iters, force_xla=False): argument174 if force_xla:214 self._test_loop_fn(loop_fn, 3, force_xla=True)241 self._test_loop_fn(loop_fn, 3, force_xla=True)