| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | array_test.py | 42 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 …]
|
| D | math_test.py | 55 def loop_fn(i): function
151 def loop_fn(i): function
223 def loop_fn(i): function
242 def loop_fn(i): function
254 def loop_fn(i): function
264 def loop_fn(i): function
291 def loop_fn(i): function
316 def loop_fn(i): function
338 def loop_fn(i): function
359 def loop_fn(i): function
[all …]
|
| D | control_flow_ops_test.py | 90 def loop_fn(i): function
103 def loop_fn(i): function
199 def loop_fn(i): function
207 def loop_fn(i): function
228 def loop_fn(i, pfor_config): function
242 def loop_fn(i, pfor_config): function
256 def loop_fn(i, pfor_config): function
268 def loop_fn(i, pfor_config): function
311 def loop_fn(i, pfor_config): function
325 def loop_fn(i, pfor_config): function
[all …]
|
| D | xla_control_flow_ops_test.py | 52 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
198 def loop_fn(_): function
209 def loop_fn(i): function
225 def loop_fn(i, pfor_config): function
|
| D | control_flow_ops.py | 46 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
|
| D | gradients.py | 53 def loop_fn(i): function
129 def loop_fn(i): function
|
| D | gradients_test.py | 176 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
|
| D | test_util.py | 52 def _test_loop_fn(self, argument
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop.py | 1162 def loop_fn(i): function
1303 def loop_fn(i): function
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_cell_test.py | 2076 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/ops/ |
| D | rnn.py | 920 loop_fn, argument
|