Searched refs:_test_loop_fn (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | array_test.py | 70 self._test_loop_fn(loop_fn, 3)
83 self._test_loop_fn(loop_fn, 3)
92 self._test_loop_fn(loop_fn, 3)
101 self._test_loop_fn(loop_fn, 3)
110 self._test_loop_fn(loop_fn, 3)
123 self._test_loop_fn(loop_fn, 3)
132 self._test_loop_fn(loop_fn, 3)
139 self._test_loop_fn(loop_fn, 3)
149 self._test_loop_fn(loop_fn, 3)
163 self._test_loop_fn(loop_fn, 3)
[all …]
|
| D | math_test.py | 72 self._test_loop_fn(loop_fn, 3)
156 self._test_loop_fn(loop_fn, 3)
233 self._test_loop_fn(loop_fn, 3)
247 self._test_loop_fn(loop_fn, 3)
258 self._test_loop_fn(loop_fn, 2)
270 self._test_loop_fn(loop_fn, 4)
298 self._test_loop_fn(loop_fn, 2)
323 self._test_loop_fn(loop_fn, 2)
344 self._test_loop_fn(loop_fn, 2)
365 self._test_loop_fn(loop_fn, 2)
[all …]
|
| D | control_flow_ops_test.py | 95 self._test_loop_fn(loop_fn, 3, fallback_to_while_loop=False)
96 self._test_loop_fn(loop_fn, 3, fallback_to_while_loop=True)
108 self._test_loop_fn(loop_fn, 8, parallel_iterations=parallel_iterations)
109 self._test_loop_fn(
203 self._test_loop_fn(loop_fn, 2)
214 self._test_loop_fn(loop_fn, 2)
348 self._test_loop_fn(loop_fn, 3)
374 self._test_loop_fn(loop_fn, 3)
387 self._test_loop_fn(loop_fn, 3)
396 self._test_loop_fn(loop_fn, 3)
[all …]
|
| D | xla_control_flow_ops_test.py | 65 self._test_loop_fn(loop_fn, num_loop)
159 def _test_loop_fn(self, loop_fn, iters, force_xla=False): member in WhileV2Test
193 self._test_loop_fn(loop_fn, 3)
205 self._test_loop_fn(loop_fn, 3)
214 self._test_loop_fn(loop_fn, 3, force_xla=True)
241 self._test_loop_fn(loop_fn, 3, force_xla=True)
|
| D | test_util.py | 52 def _test_loop_fn(self, member in PForTestCase
|