Home
last modified time | relevance | path

Searched refs:parallel_iterations (Results 1 – 25 of 135) sorted by relevance

123456

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/
H A Dcontrol_flow_ops.py45 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 Dgradients.py24 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 Dtest_util.py51 parallel_iterations=None, argument
59 parallel_iterations=parallel_iterations)
62 parallel_iterations=parallel_iterations)
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dmap_fn.py47 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 Dfunctional_ops.py49 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 Drnn.py353 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 Dcontrol_flow_ops.py223 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 Dcontrol_flow_state.py180 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 Dwhile_v2.py75 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 Dimmutable_executor_state.cc169 int parallel_iterations; in Initialize() local
171 GetNodeAttr(n->attrs(), "parallel_iterations", &parallel_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 Dlower_while_op.cc66 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 Daccumulate_n_optimizer.cc87 int parallel_iterations; in Run() local
89 &parallel_iterations); in Run()
90 if (found && parallel_iterations == 1) { in Run()
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dbackprop.py1135 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 Dwhile_op.mlir10 …_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 Dfunc_use_fallback_tensor.mlir70 … = @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 Denter.mlir8 …) 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 Dwhile_op.mlir18 …= @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 Dwhile_op.mlir18 …= @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 Dloops_and_resource_vars.pbtxt150 key: "parallel_iterations"
179 key: "parallel_iterations"
379 key: "parallel_iterations"
420 key: "parallel_iterations"
449 key: "parallel_iterations"
H A Dloops_and_queues.pbtxt112 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 Dconcrete-ops.mlir77 // 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 Ddirectives.py47 parallel_iterations=UNSPECIFIED, argument
91 del parallel_iterations
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Dragged_map_ops.py30 parallel_iterations=None, argument
164 parallel_iterations,
/aosp_15_r20/external/tensorflow/tensorflow/core/debug/
H A Ddebug_graph_utils.cc260 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 Dcontrol_flow_ops_py_test.py1556 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 …]

123456