| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | gradients.py | 28 def jacobian(output, inputs, use_pfor=True, parallel_iterations=None): argument
64 loop_fn, output_size, parallel_iterations=parallel_iterations)
70 parallel_iterations=parallel_iterations)
83 def batch_jacobian(output, inp, use_pfor=True, parallel_iterations=None): argument
135 parallel_iterations=parallel_iterations)
140 parallel_iterations=parallel_iterations)
|
| D | control_flow_ops.py | 46 def for_loop(loop_fn, loop_fn_dtypes, iters, parallel_iterations=None): argument
89 if parallel_iterations is not None:
90 extra_args = {"parallel_iterations": parallel_iterations}
138 def pfor(loop_fn, iters, fallback_to_while_loop=True, parallel_iterations=None): argument
191 parallel_iterations=parallel_iterations)
269 parallel_iterations=None, argument
310 if parallel_iterations is not None:
311 if parallel_iterations < 1:
313 if parallel_iterations == 1:
315 if iters_value is not None and iters_value < parallel_iterations:
[all …]
|
| D | test_util.py | 55 parallel_iterations=None, argument
63 parallel_iterations=parallel_iterations)
66 parallel_iterations=parallel_iterations)
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | rnn.py | 50 parallel_iterations=None, argument
220 parallel_iterations = parallel_iterations or 32
266 parallel_iterations=parallel_iterations,
289 parallel_iterations=None, argument
379 parallel_iterations=parallel_iterations,
418 parallel_iterations=parallel_iterations,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | map_fn.py | 50 parallel_iterations=None, argument
364 if in_graph_mode and not parallel_iterations:
365 parallel_iterations = 10
366 elif not in_graph_mode and not parallel_iterations:
367 parallel_iterations = 1
368 elif not in_graph_mode and parallel_iterations > 1:
374 parallel_iterations = 1
504 parallel_iterations=parallel_iterations,
634 parallel_iterations=None, argument
647 parallel_iterations=parallel_iterations,
|
| D | functional_ops.py | 53 parallel_iterations=10, argument
153 parallel_iterations=parallel_iterations,
180 parallel_iterations=10, argument
237 parallel_iterations=parallel_iterations,
248 parallel_iterations=10, argument
349 parallel_iterations=parallel_iterations,
376 parallel_iterations=10, argument
433 parallel_iterations=parallel_iterations,
444 parallel_iterations=10, argument
667 parallel_iterations=parallel_iterations,
[all …]
|
| D | rnn.py | 354 parallel_iterations=None, argument
444 parallel_iterations=parallel_iterations,
483 parallel_iterations=parallel_iterations,
510 parallel_iterations=None, argument
642 parallel_iterations = parallel_iterations or 32
688 parallel_iterations=parallel_iterations,
706 parallel_iterations, argument
745 assert isinstance(parallel_iterations, int), "parallel_iterations must be int"
892 parallel_iterations=parallel_iterations,
921 parallel_iterations=None, argument
[all …]
|
| D | control_flow_ops.py | 225 parallel_iterations=10, argument
252 data, frame_name, is_constant, parallel_iterations, name=name)
255 data, frame_name, is_constant, parallel_iterations, name=name)
262 return _Enter(t, frame_name, is_constant, parallel_iterations, use_ref,
1458 parallel_iterations=10, argument
1483 self._init_from_args(maximum_iterations, parallel_iterations, back_prop,
1488 def _init_from_args(self, maximum_iterations, parallel_iterations, back_prop, argument
1502 if not isinstance(parallel_iterations, int) or (parallel_iterations <= 0):
1504 "%s" % parallel_iterations)
1507 self._parallel_iterations = parallel_iterations
[all …]
|
| D | control_flow_state.py | 184 parallel_iterations=forward_ctxt.parallel_iterations,
198 parallel_iterations=forward_ctxt.parallel_iterations,
431 parallel_iterations = self.grad_context.parallel_iterations
432 if parallel_iterations > 1:
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | immutable_executor_state.cc | 168 int parallel_iterations; in Initialize() local
170 GetNodeAttr(n->attrs(), "parallel_iterations", ¶llel_iterations)); in Initialize()
172 if (frame_info->parallel_iterations == -1) { in Initialize()
173 frame_info->parallel_iterations = parallel_iterations; in Initialize()
174 } else if (frame_info->parallel_iterations != parallel_iterations) { in Initialize()
177 << frame_info->parallel_iterations << " vs. " in Initialize()
178 << parallel_iterations << "."; in Initialize()
|
| D | immutable_executor_state.h | 51 parallel_iterations(-1) {} in FrameInfo()
74 int32 parallel_iterations; member
|
| D | lower_while_op.cc | 66 const NameAttrList& body_fn, int parallel_iterations, in Run() argument
68 LowerWhileHelper helper(while_op, cond_fn, body_fn, parallel_iterations, in Run()
78 const NameAttrList& body_fn, int parallel_iterations,
179 int parallel_iterations, Graph* graph, in LowerWhileHelper() argument
184 parallel_iterations_(parallel_iterations), in LowerWhileHelper()
|
| D | accumulate_n_optimizer.cc | 87 int parallel_iterations; in Run() local
89 ¶llel_iterations); in Run()
90 if (found && parallel_iterations == 1) { in Run()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/executor_tpuv1_island_coarsening/ |
| D | while_op.mlir | 10 …_attr, cond = @while_cond_with_cluster_attr, is_stateless = false, parallel_iterations = 10 : i64}…
11 … cond = @while_cond_with_wrong_cluster_attr, is_stateless = false, parallel_iterations = 10 : i64}…
12 …_attr, cond = @while_cond_with_cluster_attr, is_stateless = false, parallel_iterations = 10 : i64}…
13 …tr, cond = @while_cond_without_cluster_attr, is_stateless = false, parallel_iterations = 10 : i64}…
14 …tr, cond = @while_cond_without_cluster_attr, is_stateless = false, parallel_iterations = 10 : i64}…
|
| /external/tensorflow/tensorflow/python/autograph/lang/ |
| D | directives.py | 51 parallel_iterations=UNSPECIFIED, argument
95 del parallel_iterations
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_map_ops.py | 33 parallel_iterations=None, argument
167 parallel_iterations,
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop.py | 1094 parallel_iterations=None, argument
1176 parallel_iterations=parallel_iterations)
1193 parallel_iterations=parallel_iterations)
1210 parallel_iterations=None, argument
1312 parallel_iterations=parallel_iterations)
1328 parallel_iterations=parallel_iterations)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/executor_tpuv1_outline_island/ |
| D | while_op.mlir | 18 …= @while_cond_with_cluster_attr, is_stateless = false, name = "A", parallel_iterations = 10 : i64}…
19 …= @while_cond_with_cluster_attr, is_stateless = false, name = "C", parallel_iterations = 10 : i64}…
20 …while_cond_without_cluster_attr, is_stateless = false, name = "E", parallel_iterations = 10 : i64}…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/executor_tpuv1_island_inlining/ |
| D | while_op.mlir | 18 …= @while_cond_with_cluster_attr, is_stateless = false, name = "A", parallel_iterations = 10 : i64}…
19 …= @while_cond_with_cluster_attr, is_stateless = false, name = "C", parallel_iterations = 10 : i64}…
20 …while_cond_without_cluster_attr, is_stateless = false, name = "E", parallel_iterations = 10 : i64}…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-gradient-tape.pbtxt | 11 …gs=[\'self\', \'target\', \'source\', \'unconnected_gradients\', \'parallel_iterations\', \'experi…
19 …s=[\'self\', \'target\', \'sources\', \'unconnected_gradients\', \'parallel_iterations\', \'experi…
|
| D | tensorflow.autodiff.-gradient-tape.pbtxt | 11 …gs=[\'self\', \'target\', \'source\', \'unconnected_gradients\', \'parallel_iterations\', \'experi…
19 …s=[\'self\', \'target\', \'sources\', \'unconnected_gradients\', \'parallel_iterations\', \'experi…
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_RefEnter.pbtxt | 29 name: "parallel_iterations"
38 it may be changed in the child frame. At most `parallel_iterations` iterations
|
| D | api_def_Enter.pbtxt | 29 name: "parallel_iterations"
39 it may be changed in the child frame. At most `parallel_iterations` iterations
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-gradient-tape.pbtxt | 11 …gs=[\'self\', \'target\', \'source\', \'unconnected_gradients\', \'parallel_iterations\', \'experi…
19 …s=[\'self\', \'target\', \'sources\', \'unconnected_gradients\', \'parallel_iterations\', \'experi…
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | constant_folding_test.py | 52 parallel_iterations=1)
66 parallel_iterations=1)
|