| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/ |
| H A D | control_flow_ops.py | 45 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}
141 parallel_iterations=None, argument
197 parallel_iterations=parallel_iterations,
276 parallel_iterations=None, argument
319 if parallel_iterations is not None:
320 if parallel_iterations < 1:
324 if parallel_iterations == 1:
327 if iters_value is not None and iters_value < parallel_iterations:
[all …]
|
| H A D | gradients.py | 24 def jacobian(output, inputs, use_pfor=True, parallel_iterations=None): argument
60 loop_fn, output_size, parallel_iterations=parallel_iterations)
66 parallel_iterations=parallel_iterations)
79 def batch_jacobian(output, inp, use_pfor=True, parallel_iterations=None): argument
131 parallel_iterations=parallel_iterations)
136 parallel_iterations=parallel_iterations)
|
| H A D | test_util.py | 51 parallel_iterations=None, argument
59 parallel_iterations=parallel_iterations)
62 parallel_iterations=parallel_iterations)
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | map_fn.py | 47 parallel_iterations=None, argument
362 if in_graph_mode and not parallel_iterations:
363 parallel_iterations = 10
364 elif not in_graph_mode and not parallel_iterations:
365 parallel_iterations = 1
366 elif not in_graph_mode and parallel_iterations > 1:
372 parallel_iterations = 1
501 parallel_iterations=parallel_iterations,
631 parallel_iterations=None, argument
644 parallel_iterations=parallel_iterations,
|
| H A D | functional_ops.py | 49 parallel_iterations=10, argument
150 parallel_iterations=parallel_iterations,
177 parallel_iterations=10, argument
234 parallel_iterations=parallel_iterations,
245 parallel_iterations=10, argument
347 parallel_iterations=parallel_iterations,
374 parallel_iterations=10, argument
431 parallel_iterations=parallel_iterations,
442 parallel_iterations=10, argument
666 parallel_iterations=parallel_iterations,
[all …]
|
| H A D | rnn.py | 353 parallel_iterations=None, argument
443 parallel_iterations=parallel_iterations,
482 parallel_iterations=parallel_iterations,
509 parallel_iterations=None, argument
702 parallel_iterations = parallel_iterations or 32
750 parallel_iterations=parallel_iterations,
768 parallel_iterations, argument
807 assert isinstance(parallel_iterations, int), "parallel_iterations must be int"
956 parallel_iterations=parallel_iterations,
985 parallel_iterations=None, argument
[all …]
|
| H A D | control_flow_ops.py | 223 parallel_iterations=10, argument
254 tensor, frame_name, is_constant, parallel_iterations, name=name)
257 tensor, frame_name, is_constant, parallel_iterations, name=name)
264 return _Enter(t, frame_name, is_constant, parallel_iterations, use_ref,
1416 parallel_iterations=10, argument
1441 self._init_from_args(maximum_iterations, parallel_iterations, back_prop,
1446 def _init_from_args(self, maximum_iterations, parallel_iterations, back_prop, argument
1460 if not isinstance(parallel_iterations, int) or (parallel_iterations <= 0):
1462 "%s" % parallel_iterations)
1465 self._parallel_iterations = parallel_iterations
[all …]
|
| H A D | control_flow_state.py | 180 parallel_iterations=forward_ctxt.parallel_iterations,
194 parallel_iterations=forward_ctxt.parallel_iterations,
427 parallel_iterations = self.grad_context.parallel_iterations
428 if parallel_iterations > 1:
|
| H A D | while_v2.py | 75 parallel_iterations=10, argument
312 parallel_iterations=parallel_iterations,
357 parallel_iterations = op.get_attr("parallel_iterations")
458 parallel_iterations=parallel_iterations,
469 parallel_iterations, name, num_original_outputs, argument
489 parallel_iterations=parallel_iterations,
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | immutable_executor_state.cc | 169 int parallel_iterations; in Initialize() local
171 GetNodeAttr(n->attrs(), "parallel_iterations", ¶llel_iterations)); in Initialize()
173 if (frame_info->parallel_iterations == -1) { in Initialize()
174 frame_info->parallel_iterations = parallel_iterations; in Initialize()
175 } else if (frame_info->parallel_iterations != parallel_iterations) { in Initialize()
178 << frame_info->parallel_iterations << " vs. " in Initialize()
179 << parallel_iterations << "."; in Initialize()
|
| H A D | lower_while_op.cc | 66 const NameAttrList& body_fn, int parallel_iterations, in Run() argument
69 LowerWhileHelper helper(while_op, cond_fn, body_fn, parallel_iterations, in Run()
79 const NameAttrList& body_fn, int parallel_iterations,
182 int parallel_iterations, Graph* graph, in LowerWhileHelper() argument
189 parallel_iterations_(parallel_iterations), in LowerWhileHelper()
|
| H A 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | backprop.py | 1135 parallel_iterations=None, argument
1217 parallel_iterations=parallel_iterations)
1231 parallel_iterations=parallel_iterations)
1248 parallel_iterations=None, argument
1346 if parallel_iterations is not None:
1364 parallel_iterations=parallel_iterations)
1377 parallel_iterations=parallel_iterations)
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/executor_tpuv1_island_coarsening/ |
| H A 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}…
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/ |
| H A D | func_use_fallback_tensor.mlir | 70 … = @while_cond_lt9, body = @while_body_add2, is_stateless = false, parallel_iterations = 1} : (ten…
91 // CHECK-SAME: parallel_iterations(10)
94 // CHECK-SAME: parallel_iterations(1)
95 … = @while_cond_lt9, body = @while_body_add2, is_stateless = false, parallel_iterations = 10} : (te…
96 … = @while_cond_lt9, body = @while_body_add2, is_stateless = false, parallel_iterations = 1} : (ten…
120 …nd_lt9, body = @side_effect_while_body_add2, is_stateless = false, parallel_iterations = 1} : (ten…
150 // CHECK-SAME: parallel_iterations(10)
151 …ond = @tensor_array_while_cond, device = "", is_stateless = false, parallel_iterations = 10 : i64}…
159 // CHECK-SAME: parallel_iterations(10)
160 …ond = @tensor_array_while_cond, device = "", is_stateless = false, parallel_iterations = 10 : i64}…
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/tests/ |
| H A D | enter.mlir | 8 …) name("enter1") {T = f32, frame_name = "foo", is_constant = true, parallel_iterations = 10 : i64}…
11 …) name("enter2") {T = f32, frame_name = "foo", is_constant = true, parallel_iterations = 10 : i64}…
12 … name("enter3") {T = f32, frame_name = "foo", is_constant = false, parallel_iterations = 10 : i64}…
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/executor_tpuv1_outline_island/ |
| H A 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}…
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/executor_tpuv1_island_inlining/ |
| H A 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}…
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/graph_properties_testdata/ |
| H A D | loops_and_resource_vars.pbtxt | 150 key: "parallel_iterations"
179 key: "parallel_iterations"
379 key: "parallel_iterations"
420 key: "parallel_iterations"
449 key: "parallel_iterations"
|
| H A D | loops_and_queues.pbtxt | 112 key: "parallel_iterations"
141 key: "parallel_iterations"
307 key: "parallel_iterations"
445 key: "parallel_iterations"
474 key: "parallel_iterations"
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ir/tests/ |
| H A D | concrete-ops.mlir | 77 // CHECK-SAME: parallel_iterations = 10 : i64
78 parallel_iterations = 10 : i64
144 // CHECK-SAME: parallel_iterations = 10 : i64
145 parallel_iterations = 10 : i64
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/autograph/lang/ |
| H A D | directives.py | 47 parallel_iterations=UNSPECIFIED, argument
91 del parallel_iterations
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_map_ops.py | 30 parallel_iterations=None, argument
164 parallel_iterations,
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/debug/ |
| H A D | debug_graph_utils.cc | 260 const AttrValue* parallel_iterations = in DeparallelizeWhileLoops() local
262 if (parallel_iterations && parallel_iterations->i() > 1) { in DeparallelizeWhileLoops()
266 << device->name() << "\" from " << parallel_iterations->i() in DeparallelizeWhileLoops()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/ |
| H A D | control_flow_ops_py_test.py | 1556 r = control_flow_ops.while_loop(c, b, [n], parallel_iterations=20)
1606 r = control_flow_ops.while_loop(c, b, [i, x], parallel_iterations=5)
2057 r = control_flow_ops.while_loop(c, b, [n], parallel_iterations=20)
2065 r = control_flow_ops.while_loop(c, b, [n], parallel_iterations=20)
2396 r_s = control_flow_ops.while_loop(c, b, [s], parallel_iterations=1)
2409 outer_c, outer_b, [s0], parallel_iterations=1)
2430 condition, body, [n, r], parallel_iterations=1)
2445 condition, body, [r], parallel_iterations=1)
2684 loop_iterator, loop_body, [n], parallel_iterations=1)
2710 loop_iterator, loop_body, [n], parallel_iterations=1)
[all …]
|