| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | array_test.py | 40 def loop_fn(i): function
53 self._test_loop_fn(loop_fn, 3, loop_fn_dtypes=[dtypes.float32] * 20)
58 def loop_fn(i): function
62 self._test_loop_fn(loop_fn, 3, loop_fn_dtypes=[dtypes.int32, dtypes.int64])
67 def loop_fn(i): function
71 self._test_loop_fn(loop_fn, 3, loop_fn_dtypes=[dtypes.int32, dtypes.int64])
76 def loop_fn(i): function
80 self._test_loop_fn(loop_fn, 3, loop_fn_dtypes=[dtypes.int32])
86 def loop_fn(i): function
93 loop_fn, 3, loop_fn_dtypes=[dtypes.int32] * 4 + [dtypes.int64] * 4)
[all …]
|
| D | control_flow_ops_test.py | 68 def loop_fn(i): function
74 loop_fn, 3, loop_fn_dtypes=[dtypes.float32, dtypes.int32])
77 loop_fn, 3, loop_fn_dtypes=[dtypes.float32, dtypes.int32])
85 def loop_fn(i): function
89 self._test_loop_fn(loop_fn, 8, parallel_iterations=parallel_iterations)
90 self._test_loop_fn(loop_fn, 4 * constant_op.constant(2),
111 def loop_fn(i, pfor_config): function
117 output = pfor_control_flow_ops.pfor(loop_fn, 8)
125 def loop_fn(i, pfor_config): function
129 output = pfor_control_flow_ops.pfor(loop_fn, 8)
[all …]
|
| D | math_test.py | 102 def loop_fn(i): function
121 self._test_loop_fn(loop_fn, 3, loop_fn_dtypes=output_dtypes)
132 def loop_fn(i): function
137 self._test_loop_fn(loop_fn, 3, loop_fn_dtypes=x.dtype)
197 def loop_fn(i): function
206 self._test_loop_fn(loop_fn, 3, loop_fn_dtypes=output_dtypes)
212 def loop_fn(i): function
217 self._test_loop_fn(loop_fn, 3, loop_fn_dtypes=[dtypes.bool])
224 def loop_fn(i): function
228 self._test_loop_fn(loop_fn, 2)
[all …]
|
| D | control_flow_ops.py | 40 def for_loop(loop_fn, loop_fn_dtypes, iters, parallel_iterations=None): argument
66 fn_output = nest.flatten(loop_fn(i))
111 def pfor(loop_fn, iters, parallel_iterations=None): argument
159 return _pfor_impl(loop_fn, iters, parallel_iterations=parallel_iterations)
165 def _loop_fn_has_config(loop_fn): argument
167 if tf_inspect.isfunction(loop_fn):
168 argspec = tf_inspect.getargspec(loop_fn)
170 elif isinstance(loop_fn, functools.partial):
171 fn = loop_fn.func
174 PFOR_CONFIG_ARG not in loop_fn.keywords)
[all …]
|
| D | gradients.py | 51 def loop_fn(i): function
62 loop_fn, output_size, parallel_iterations=parallel_iterations)
65 loop_fn,
124 def loop_fn(i): function
129 pfor_output = control_flow_ops.pfor(loop_fn, output_row_size,
133 loop_fn, output.dtype,
|
| D | gradients_test.py | 177 def loop_fn(i): function
180 pfor_outputs = control_flow_ops.pfor(loop_fn, batch_size)
182 while_outputs = control_flow_ops.for_loop(loop_fn, loop_fn_dtypes, batch_size)
201 def loop_fn(i): function
209 pfor_outputs = control_flow_ops.pfor(loop_fn, batch_size)
211 while_outputs = control_flow_ops.for_loop(loop_fn, loop_fn_dtypes, batch_size)
270 def loop_fn(i): function
274 pfor_outputs = control_flow_ops.pfor(loop_fn, batch_size)
276 loop_fn, dtypes.float32, batch_size)
289 def loop_fn(i): function
[all …]
|
| D | test_util.py | 52 def _test_loop_fn(self, loop_fn, iters, argument
55 t1 = pfor_control_flow_ops.pfor(loop_fn, iters=iters,
57 t2 = pfor_control_flow_ops.for_loop(loop_fn, loop_fn_dtypes, iters=iters,
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop.py | 1021 def loop_fn(i): function
1035 output = pfor_ops.pfor(loop_fn, target_size,
1052 loop_fn, [target.dtype] * len(flat_sources), target_size,
1149 def loop_fn(i): function
1158 output = pfor_ops.pfor(loop_fn, target_row_size,
1174 output = pfor_ops.for_loop(loop_fn, target.dtype, target_row_size,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn.py | 902 def raw_rnn(cell, loop_fn, argument
1073 if not callable(loop_fn):
1088 init_loop_state) = loop_fn(
1170 next_loop_state) = loop_fn(
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_cell_test.py | 2066 def loop_fn(time_, cell_output, cell_state, unused_loop_state): function
2083 outputs_ta, final_state, _ = rnn.raw_rnn(cell, loop_fn, scope=reuse_scope)
2175 def loop_fn(time_, cell_output, cell_state, loop_state): function
2193 r = rnn.raw_rnn(cell, loop_fn)
2212 def loop_fn(time_, cell_output, cell_state, loop_state): function
2236 r = rnn.raw_rnn(cell, loop_fn)
2259 def loop_fn(time_, cell_output, cell_state, _): function
2278 r = rnn.raw_rnn(cell, loop_fn)
2328 def loop_fn(time_, cell_output, cell_state, unused_loop_state): function
2344 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…
|