| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | array_test.py | 42 def loop_fn(i): function
70 self._test_loop_fn(loop_fn, 3)
75 def loop_fn(i): function
83 self._test_loop_fn(loop_fn, 3)
88 def loop_fn(i): function
92 self._test_loop_fn(loop_fn, 3)
97 def loop_fn(i): function
101 self._test_loop_fn(loop_fn, 3)
106 def loop_fn(i): function
110 self._test_loop_fn(loop_fn, 3)
[all …]
|
| D | control_flow_ops_test.py | 90 def loop_fn(i): function
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)
103 def loop_fn(i): function
108 self._test_loop_fn(loop_fn, 8, parallel_iterations=parallel_iterations)
110 loop_fn,
199 def loop_fn(i): function
203 self._test_loop_fn(loop_fn, 2)
207 def loop_fn(i): function
214 self._test_loop_fn(loop_fn, 2)
[all …]
|
| D | math_test.py | 55 def loop_fn(i): function
72 self._test_loop_fn(loop_fn, 3)
151 def loop_fn(i): function
156 self._test_loop_fn(loop_fn, 3)
223 def loop_fn(i): function
233 self._test_loop_fn(loop_fn, 3)
242 def loop_fn(i): function
247 self._test_loop_fn(loop_fn, 3)
254 def loop_fn(i): function
258 self._test_loop_fn(loop_fn, 2)
[all …]
|
| D | control_flow_ops.py | 46 def for_loop(loop_fn, loop_fn_dtypes, iters, parallel_iterations=None): argument
72 fn_output = nest.flatten(loop_fn(i))
138 def pfor(loop_fn, iters, fallback_to_while_loop=True, parallel_iterations=None): argument
188 return _pfor_impl(loop_fn,
248 def _loop_fn_has_config(loop_fn): argument
250 if tf_inspect.isfunction(loop_fn):
251 argspec = tf_inspect.getargspec(loop_fn)
253 elif isinstance(loop_fn, functools.partial):
254 fn = loop_fn.func
257 PFOR_CONFIG_ARG not in loop_fn.keywords)
[all …]
|
| D | xla_control_flow_ops_test.py | 52 def loop_fn(i): function
65 self._test_loop_fn(loop_fn, num_loop)
118 def loop_fn(i, pfor_config): function
122 return pfor_control_flow_ops.pfor(loop_fn, 8)
159 def _test_loop_fn(self, loop_fn, iters, force_xla=False): argument
162 return pfor_control_flow_ops.pfor(loop_fn, iters)
184 def loop_fn(i): function
193 self._test_loop_fn(loop_fn, 3)
198 def loop_fn(_): function
205 self._test_loop_fn(loop_fn, 3)
[all …]
|
| D | gradients.py | 53 def loop_fn(i): function
64 loop_fn, output_size, parallel_iterations=parallel_iterations)
67 loop_fn,
129 def loop_fn(i): function
134 pfor_output = control_flow_ops.pfor(loop_fn, output_row_size,
138 loop_fn, output.dtype,
|
| D | gradients_test.py | 176 def loop_fn(i): function
179 pfor_outputs = control_flow_ops.pfor(loop_fn, batch_size)
181 while_outputs = control_flow_ops.for_loop(loop_fn, loop_fn_dtypes, batch_size)
200 def loop_fn(i): function
208 pfor_outputs = control_flow_ops.pfor(loop_fn, batch_size)
210 while_outputs = control_flow_ops.for_loop(loop_fn, loop_fn_dtypes, batch_size)
269 def loop_fn(i): function
273 pfor_outputs = control_flow_ops.pfor(loop_fn, batch_size)
275 loop_fn, dtypes.float32, batch_size)
288 def loop_fn(i): function
[all …]
|
| D | test_util.py | 53 loop_fn,
60 loop_fn,
65 t2 = pfor_control_flow_ops.for_loop(loop_fn, loop_fn_dtypes, iters=iters,
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop.py | 1162 def loop_fn(i): function
1175 output = pfor_ops.pfor(loop_fn, target_size,
1192 loop_fn, [target.dtype] * len(flat_sources), target_size,
1303 def loop_fn(i): function
1311 output = pfor_ops.pfor(loop_fn, target_row_size,
1327 output = pfor_ops.for_loop(loop_fn, target.dtype, target_row_size,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn.py | 920 loop_fn, argument
1086 if not callable(loop_fn):
1101 initial_state, emit_structure, init_loop_state) = loop_fn(
1187 next_loop_state) = loop_fn(next_time, next_output, cell_state,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_cell_test.py | 2076 def loop_fn(time_, cell_output, cell_state, unused_loop_state): function
2093 outputs_ta, final_state, _ = rnn.raw_rnn(cell, loop_fn, scope=reuse_scope)
2185 def loop_fn(time_, cell_output, cell_state, loop_state): function
2203 r = rnn.raw_rnn(cell, loop_fn)
2222 def loop_fn(time_, cell_output, cell_state, loop_state): function
2246 r = rnn.raw_rnn(cell, loop_fn)
2269 def loop_fn(time_, cell_output, cell_state, _): function
2288 r = rnn.raw_rnn(cell, loop_fn)
2338 def loop_fn(time_, cell_output, cell_state, unused_loop_state): function
2354 return rnn.raw_rnn(cell, loop_fn, scope=scope)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.nn.pbtxt | 313 …argspec: "args=[\'cell\', \'loop_fn\', \'parallel_iterations\', \'swap_memory\', \'scope\'], varar…
|