| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| 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)
|
| 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 …]
|
| D | test_util.py | 51 parallel_iterations=None, argument
59 parallel_iterations=parallel_iterations)
62 parallel_iterations=parallel_iterations)
|
| /external/tensorflow/tensorflow/python/ops/ |
| 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,
|
| 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 …]
|
| 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 …]
|
| 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 …]
|
| 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:
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| 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()
|
| 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
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()
|
| /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 | 47 parallel_iterations=UNSPECIFIED, argument
91 del parallel_iterations
|
| /external/tensorflow/tensorflow/core/transforms/constant_folding/tests/ |
| 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}…
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_map_ops.py | 30 parallel_iterations=None, argument
164 parallel_iterations,
|
| /external/tensorflow/tensorflow/python/eager/ |
| 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)
|
| /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/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/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/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/compiler/mlir/tfrt/tests/tf_to_corert/ |
| 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}…
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | constant_folding_test.py | 48 parallel_iterations=1)
62 parallel_iterations=1)
|