Home
last modified time | relevance | path

Searched defs:loop_fn (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/python/ops/parallel_for/
Darray_test.py42 def loop_fn(i): function
75 def loop_fn(i): function
88 def loop_fn(i): function
97 def loop_fn(i): function
106 def loop_fn(i): function
116 def loop_fn(i): function
128 def loop_fn(i): function
136 def loop_fn(i): function
144 def loop_fn(i): function
154 def loop_fn(i): function
[all …]
Dmath_test.py56 def loop_fn(i): function
147 def loop_fn(i): function
219 def loop_fn(i): function
238 def loop_fn(i): function
253 def loop_fn(i): function
265 def loop_fn(i): function
275 def loop_fn(i): function
302 def loop_fn(i): function
327 def loop_fn(i): function
349 def loop_fn(i): function
[all …]
Dcontrol_flow_ops_test.py96 def loop_fn(i): function
109 def loop_fn(i): function
216 def loop_fn(i): function
224 def loop_fn(i): function
245 def loop_fn(i, pfor_config): function
259 def loop_fn(i, pfor_config): function
273 def loop_fn(i, pfor_config): function
285 def loop_fn(i, pfor_config): function
328 def loop_fn(i, pfor_config): function
342 def loop_fn(i, pfor_config): function
[all …]
Dxla_control_flow_ops_test.py52 def loop_fn(i): function
118 def loop_fn(i, pfor_config): function
159 def _test_loop_fn(self, loop_fn, iters, force_xla=False): argument
184 def loop_fn(i): function
201 def loop_fn(_): function
212 def loop_fn(i): function
228 def loop_fn(i, pfor_config): function
Dcontrol_flow_ops.py46 def for_loop(loop_fn, loop_fn_dtypes, iters, parallel_iterations=None): argument
138 def pfor(loop_fn, iters, fallback_to_while_loop=True, parallel_iterations=None): argument
248 def _loop_fn_has_config(loop_fn): argument
266 def _pfor_impl(loop_fn, argument
524 def loop_fn(i): function
Dgradients.py53 def loop_fn(i): function
129 def loop_fn(i): function
Dgradients_test.py176 def loop_fn(i): function
200 def loop_fn(i): function
269 def loop_fn(i): function
288 def loop_fn(i): function
317 def loop_fn(i, use_pfor): function
339 def loop_fn(i, use_pfor): function
Dtest_util.py52 def _test_loop_fn(self, argument
/external/tensorflow/tensorflow/python/kernel_tests/
Drnn_cell_test.py2076 def loop_fn(time_, cell_output, cell_state, unused_loop_state): function
2185 def loop_fn(time_, cell_output, cell_state, loop_state): function
2222 def loop_fn(time_, cell_output, cell_state, loop_state): function
2269 def loop_fn(time_, cell_output, cell_state, _): function
2338 def loop_fn(time_, cell_output, cell_state, unused_loop_state): function
/external/tensorflow/tensorflow/python/eager/
Dbackprop.py1172 def loop_fn(i): function
1315 def loop_fn(i): function
/external/tensorflow/tensorflow/python/ops/
Drnn.py981 loop_fn, argument