| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | control_flow_deprecated_py2.py | 402 def while_body(iterate_index, *loop_vars): argument
405 new_vars = body(iterate, *loop_vars)
407 loop_vars = (iterate_index + 1,)
409 loop_vars += new_vars
411 return loop_vars
413 def while_cond(iterate_index, *loop_vars): argument
416 lambda: extra_test(*loop_vars),
466 def while_body(iterate_index, *loop_vars): argument
469 new_vars = body(iterate, *loop_vars)
471 loop_vars = (iterate_index + 1,)
[all …]
|
| D | control_flow_test.py | 71 def set_state(loop_vars): argument
73 s, = loop_vars
91 def set_state(loop_vars): argument
93 s, = loop_vars
113 def set_state(loop_vars): argument
115 s, = loop_vars
135 def set_state(loop_vars): argument
137 s, = loop_vars
157 def set_state(loop_vars): argument
159 s, = loop_vars
[all …]
|
| D | control_flow.py | 500 iterate_index, *loop_vars = aug_loop_vars # pylint:disable=unused-variable
503 set_state(loop_vars)
548 iterate_index, *loop_vars = aug_loop_vars # pylint:disable=unused-variable
551 set_state(loop_vars)
597 iterate, *loop_vars = aug_loop_vars # pylint:disable=unused-variable
600 set_state(loop_vars)
649 has_next, *loop_vars = aug_loop_vars # pylint:disable=unused-variable
650 set_state(loop_vars)
660 loop_vars = aug_get_state() # updated by set_state() in _tf_while_loop.
669 init_vars, loop_vars, new_loop_vars, symbol_names, opts)
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | while_v2_indexed_slices_rewriter.py | 31 def rewrite_grad_indexed_slices(grads, body_grad_graph, loop_vars, argument
76 loop_vars = _rewrite_input_as_indexed_slices(body_grad_graph, output,
77 forward_input, loop_vars)
81 return loop_vars
110 forward_input, loop_vars): argument
163 body_grad_graph, loop_vars, init_slices, input_slices, new_output_slices,
248 def _update_indexed_slices_param(graph, loop_vars, init_slices, input_slices, argument
287 return loop_vars[:flat_idx] + _flatten(init_slices) + loop_vars[flat_idx + 1:]
|
| D | while_v2.py | 62 loop_vars, argument
71 orig_loop_vars = loop_vars
78 loop_vars = list(_tensor_array_to_flow(orig_loop_vars))
79 loop_vars = nest.map_structure(
80 ops.internal_convert_to_tensor_or_indexed_slices, loop_vars,
86 control_flow_ops._shape_invariant_to_type_spec, loop_vars,
89 control_flow_ops._get_shape_invariant, loop_vars,
94 type_spec.type_spec_from_value, loop_vars, expand_composites=False)
96 control_flow_ops._get_shape_invariant, loop_vars,
113 loop_vars = [loop_counter, maximum_iterations_loop_var] + loop_vars
[all …]
|
| D | control_flow_ops.py | 2150 def _BuildLoop(self, pred, body, original_loop_vars, loop_vars, argument
2157 self._InitializeValues(loop_vars)
2158 real_vars = loop_vars
2160 real_vars = [self._outer_context.AddValue(x) for x in loop_vars]
2275 def BuildLoop(self, pred, body, loop_vars, shape_invariants, argument
2280 original_loop_vars = loop_vars
2282 loop_vars = nest.map_structure(
2284 nest.flatten(loop_vars, expand_composites=False),
2286 loop_vars = ops.convert_n_to_tensor_or_composite(loop_vars)
2289 _get_shape_invariant, loop_vars, expand_composites=False)
[all …]
|
| D | gradients_impl.py | 423 loop_vars = [
433 loop_vars
|
| D | collective_ops_test.py | 196 loop_vars = []
199 loop_vars.append(
202 loop_vars.append(variables.VariableV1(0.))
228 loop_vars))
|
| /external/tensorflow/tensorflow/python/autograph/converters/ |
| D | control_flow_deprecated_py2.py | 392 loop_vars = tuple(basic_loop_vars)
393 loop_vars_ast_tuple = gast.Tuple([n.ast() for n in loop_vars], None)
395 if len(loop_vars) == 1:
396 loop_vars = loop_vars[0]
398 return loop_vars, loop_vars_ast_tuple
407 loop_vars, loop_vars_ast_tuple = self._loop_var_constructs(
425 if loop_vars:
445 loop_vars=loop_vars,
497 loop_vars, loop_vars_ast_tuple = self._loop_var_constructs(
517 loop_vars=loop_vars,
[all …]
|
| D | control_flow.py | 266 loop_vars, undefined, _ = self._get_block_vars(node, body_scope.bound)
270 nonlocal_declarations = self._create_nonlocal_declarations(loop_vars)
276 loop_vars, nonlocal_declarations, state_getter_name, state_setter_name)
305 symbol_names=tuple(gast.Constant(str(s), kind=None) for s in loop_vars),
317 loop_vars, undefined, _ = self._get_block_vars(
322 nonlocal_declarations = self._create_nonlocal_declarations(loop_vars)
328 loop_vars, nonlocal_declarations, state_getter_name, state_setter_name)
348 loop_vars=loop_vars,
392 symbol_names=tuple(gast.Constant(str(s), kind=None) for s in loop_vars),
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | executor_test.cc | 546 int loop_iters, int loop_vars, bool lower) { in BM_WhileLoopHelper() argument
556 args.reserve(loop_vars); in BM_WhileLoopHelper()
558 for (int i = 1; i < loop_vars; ++i) { in BM_WhileLoopHelper()
563 body_rets.reserve(loop_vars); in BM_WhileLoopHelper()
565 for (int i = 1; i < loop_vars; ++i) { in BM_WhileLoopHelper()
570 body_nodes.reserve(1 + loop_vars); in BM_WhileLoopHelper()
574 for (int i = 1; i < loop_vars; ++i) { in BM_WhileLoopHelper()
615 std::vector<DataType> input_types(loop_vars, DT_INT32); in BM_WhileLoopHelper()
616 inputs.reserve(loop_vars); in BM_WhileLoopHelper()
617 for (int i = 0; i < loop_vars; ++i) { in BM_WhileLoopHelper()
[all …]
|
| /external/tensorflow/tensorflow/python/tf_program/ |
| D | mlir_gen.py | 362 loop_vars = tuple(basic_loop_vars | composite_loop_vars)
369 return loop_vars, reserved_symbols, undefined_lives
376 loop_vars, _, _ = self._get_loop_vars(node, body_scope.modified)
377 inputs = [self.symbol_table.lookup(str(name)) for name in loop_vars]
387 for input_, type_ in zip(loop_vars, types):
398 for input_, type_ in zip(loop_vars, types):
406 [self.symbol_table.lookup(str(name)) for name in loop_vars])
410 for idx, var in enumerate(loop_vars):
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | cholesky_expander.cc | 78 auto body_fn = [&](XlaOp i, absl::Span<const XlaOp> loop_vars, in CholeskyUnblocked()
82 auto body_a = loop_vars[0]; in CholeskyUnblocked()
83 auto body_l = loop_vars[1]; in CholeskyUnblocked()
84 auto seen_error = loop_vars[2]; in CholeskyUnblocked()
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | metrics_utils.py | 433 loop_vars = {
442 loop_vars[ConfusionMatrix.FALSE_NEGATIVES] = (label_is_pos, pred_is_neg)
446 loop_vars[ConfusionMatrix.FALSE_POSITIVES] = (label_is_neg, pred_is_pos)
448 loop_vars[ConfusionMatrix.TRUE_NEGATIVES] = (label_is_neg, pred_is_neg)
450 for matrix_cond, (label, pred) in loop_vars.items():
|
| /external/icu/icu4c/source/python/icutools/databuilder/renderers/ |
| D | makefile.py | 225 for loop_vars in utils.repeated_execution_request_looper(request):
226 (_, specific_dep_files, input_file, output_file) = loop_vars
231 loop_vars,
|
| D | common_exec.py | 95 for loop_vars in utils.repeated_execution_request_looper(request):
99 loop_vars,
|
| /external/icu/icu4c/source/python/icutools/databuilder/ |
| D | utils.py | 98 def format_repeated_request_command(request, cmd_template, loop_vars, common_vars): argument
99 (iter_vars, _, input_file, output_file) = loop_vars
|
| /external/tensorflow/tensorflow/python/ops/risc/ |
| D | risc_ops.py | 270 loop_vars, argument
280 loop_vars=loop_vars,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | random_ops.cc | 170 absl::Span<const xla::XlaOp> loop_vars, in Compile()
173 auto swaps = loop_vars[0]; in Compile()
174 auto indices = loop_vars[1]; in Compile()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | tf_optimizer_test.py | 109 loop_vars = [init_buf, start]
114 buf, _ = control_flow_ops.while_loop(_Cond, _Body, loop_vars, shape_inv)
|
| D | constant_folding_test.py | 64 loop_vars=[0, init_y],
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/internal/testdata/ |
| D | gen_saved_models.py | 89 loop_vars=(constant_op.constant(0.0), value))
|
| /external/tensorflow/tensorflow/python/autograph/g3doc/reference/ |
| D | limitations.md | 182 x = tf.while_loop(..., loop_vars=(x,)
196 `loop_vars` because it is not affected by the `while` loop:
205 x = tf.while_loop(..., loop_vars=(x,) # y does not need to be a loop variable
425 loop_vars=(0,))
653 loop_vars=(tf.constant(1, dtype=tf.int32),)) # Error -- inconsistent dtypes: int32, float32
686 loop_vars=(tf.constant(1,))
|
| /external/mesa3d/src/compiler/nir/ |
| D | nir_loop_analyze.c | 68 nir_loop_variable *loop_vars; member
80 return &(state->loop_vars[value->index]); in get_loop_var()
1234 state->loop_vars = rzalloc_array(mem_ctx, nir_loop_variable, in initialize_loop_info_state()
|
| /external/tensorflow/tensorflow/python/compiler/xla/ |
| D | xla_test.py | 175 cond=lambda i: math_ops.less(i, 10), body=while_body, loop_vars=[i])
|