| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/ |
| H A D | control_flow_ops_test.py | 92 def loop_fn(i): function
105 def loop_fn(i): function
212 def loop_fn(i): function
220 def loop_fn(i): function
241 def loop_fn(i, pfor_config): function
255 def loop_fn(i, pfor_config): function
269 def loop_fn(i, pfor_config): function
281 def loop_fn(i, pfor_config): function
324 def loop_fn(i, pfor_config): function
338 def loop_fn(i, pfor_config): function
[all …]
|
| H A D | array_test.py | 39 def loop_fn(i): function
72 def loop_fn(i): function
94 def loop_fn(i): function
103 def loop_fn(i): function
112 def loop_fn(i): function
122 def loop_fn(i): function
134 def loop_fn(i): function
142 def loop_fn(i): function
150 def loop_fn(i): function
160 def loop_fn(i): function
[all …]
|
| H A D | math_test.py | 52 def loop_fn(i): function
143 def loop_fn(i): function
215 def loop_fn(i): function
234 def loop_fn(i): function
249 def loop_fn(i): function
261 def loop_fn(i): function
271 def loop_fn(i): function
298 def loop_fn(i): function
323 def loop_fn(i): function
345 def loop_fn(i): function
[all …]
|
| H A D | xla_control_flow_ops_test.py | 48 def loop_fn(i): function
114 def loop_fn(i, pfor_config): function
155 def _test_loop_fn(self, loop_fn, iters, force_xla=False): argument
180 def loop_fn(i): function
197 def loop_fn(_): function
208 def loop_fn(i): function
224 def loop_fn(i, pfor_config): function
|
| H A D | control_flow_ops.py | 45 def for_loop(loop_fn, loop_fn_dtypes, iters, parallel_iterations=None): argument
138 def pfor(loop_fn, argument
255 def _loop_fn_has_config(loop_fn): argument
273 def _pfor_impl(loop_fn, argument
543 def loop_fn(i): function
|
| H A D | gradients_test.py | 172 def loop_fn(i): function
196 def loop_fn(i): function
265 def loop_fn(i): function
284 def loop_fn(i): function
313 def loop_fn(i, use_pfor): function
335 def loop_fn(i, use_pfor): function
|
| H A D | gradients.py | 49 def loop_fn(i): function
125 def loop_fn(i): function
|
| H A D | test_util.py | 48 def _test_loop_fn(self, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| H A D | rnn_cell_test.py | 2154 def loop_fn(time_, cell_output, cell_state, unused_loop_state): function
2263 def loop_fn(time_, cell_output, cell_state, loop_state): function
2300 def loop_fn(time_, cell_output, cell_state, loop_state): function
2347 def loop_fn(time_, cell_output, cell_state, _): function
2416 def loop_fn(time_, cell_output, cell_state, unused_loop_state): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | backprop.py | 1203 def loop_fn(i): function
1343 def loop_fn(i): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/ |
| H A D | map_fn_test.py | 99 def loop_fn(x): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | rnn.py | 984 loop_fn, argument
|