| /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 | 57 def set_state(loop_vars): argument
59 s, = loop_vars
77 def set_state(loop_vars): argument
79 s, = loop_vars
97 def set_state(loop_vars): argument
99 s, = loop_vars
117 def set_state(loop_vars): argument
119 s, = loop_vars
137 def set_state(loop_vars): argument
139 s, = loop_vars
[all …]
|
| D | control_flow.py | 414 iterate_index.value, loop_vars = aug_loop_vars[0], aug_loop_vars[1:]
417 set_state(loop_vars)
460 iterate_index.value, loop_vars = aug_loop_vars[0], aug_loop_vars[1:]
463 set_state(loop_vars)
498 iterate.value, loop_vars = aug_loop_vars[0], aug_loop_vars[1:]
501 set_state(loop_vars)
538 has_next.value, loop_vars = aug_loop_vars[0], aug_loop_vars[1:]
539 set_state(loop_vars)
548 loop_vars = aug_get_state() # updated by set_state() in _tf_while_loop.
557 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
101 forward_input, loop_vars): argument
154 body_grad_graph, loop_vars, init_slices, input_slices, new_output_slices,
239 def _update_indexed_slices_param(graph, loop_vars, init_slices, input_slices, argument
275 return loop_vars[:flat_idx] + _flatten(init_slices) + loop_vars[flat_idx + 1:]
|
| D | while_v2.py | 59 loop_vars, argument
68 orig_loop_vars = loop_vars
75 loop_vars = list(_tensor_array_to_flow(orig_loop_vars))
76 loop_vars = nest.map_structure(
77 ops.internal_convert_to_tensor_or_indexed_slices, loop_vars,
83 control_flow_ops._shape_invariant_to_type_spec, loop_vars,
86 control_flow_ops._get_shape_invariant, loop_vars,
91 type_spec.type_spec_from_value, loop_vars, expand_composites=False)
93 control_flow_ops._get_shape_invariant, loop_vars,
110 loop_vars = [loop_counter, maximum_iterations_loop_var] + loop_vars
[all …]
|
| D | control_flow_ops.py | 2100 def _BuildLoop(self, pred, body, original_loop_vars, loop_vars, argument
2107 self._InitializeValues(loop_vars)
2108 real_vars = loop_vars
2110 real_vars = [self._outer_context.AddValue(x) for x in loop_vars]
2225 def BuildLoop(self, pred, body, loop_vars, shape_invariants, argument
2230 original_loop_vars = loop_vars
2232 loop_vars = nest.map_structure(
2234 nest.flatten(loop_vars, expand_composites=False),
2236 loop_vars = ops.convert_n_to_tensor_or_composite(loop_vars)
2239 _get_shape_invariant, loop_vars, expand_composites=False)
[all …]
|
| D | gradients_impl.py | 407 loop_vars = [
417 loop_vars
|
| D | collective_ops_test.py | 146 loop_vars = []
149 loop_vars.append(
152 loop_vars.append(variables.VariableV1(0.))
178 loop_vars))
|
| D | rnn.py | 913 loop_vars=(time, output_ta, state),
1248 loop_vars=[
|
| /external/tensorflow/tensorflow/python/autograph/converters/ |
| D | control_flow.py | 143 def _create_nonlocal_declarations(self, loop_vars): argument
151 v for v in loop_vars if not v.is_composite() and v not in global_vars]
158 self, loop_vars, nonlocal_declarations, getter_name, setter_name): argument
159 if loop_vars:
172 state_vars=tuple(loop_vars))
389 loop_vars = tuple(basic_loop_vars | composite_loop_vars)
396 return loop_vars, reserved_symbols, undefined_lives
402 loop_vars, reserved_symbols, possibly_undefs = self._get_loop_vars(
407 nonlocal_declarations = self._create_nonlocal_declarations(loop_vars)
412 loop_vars, nonlocal_declarations, state_getter_name, state_setter_name)
[all …]
|
| D | control_flow_deprecated_py2.py | 376 loop_vars = tuple(basic_loop_vars)
377 loop_vars_ast_tuple = gast.Tuple([n.ast() for n in loop_vars], None)
379 if len(loop_vars) == 1:
380 loop_vars = loop_vars[0]
382 return loop_vars, loop_vars_ast_tuple
391 loop_vars, loop_vars_ast_tuple = self._loop_var_constructs(
409 if loop_vars:
429 loop_vars=loop_vars,
481 loop_vars, loop_vars_ast_tuple = self._loop_var_constructs(
501 loop_vars=loop_vars,
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | cholesky_expander.cc | 78 [&](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 | 432 loop_vars = {
441 loop_vars[ConfusionMatrix.FALSE_NEGATIVES] = (label_is_pos, pred_is_neg)
445 loop_vars[ConfusionMatrix.FALSE_POSITIVES] = (label_is_neg, pred_is_pos)
447 loop_vars[ConfusionMatrix.TRUE_NEGATIVES] = (label_is_neg, pred_is_neg)
449 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 | 91 for loop_vars in utils.repeated_execution_request_looper(request):
95 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/compiler/tf2xla/kernels/ |
| D | random_ops.cc | 169 absl::Span<const xla::XlaOp> loop_vars, in Compile()
172 auto swaps = loop_vars[0]; in Compile()
173 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 | 59 loop_vars=[0, init_y],
|
| D | hierarchical_controller.py | 935 loop_vars = [
939 loop_outputs = control_flow_ops.while_loop(condition, body, loop_vars)
|
| /external/tensorflow/tensorflow/python/autograph/g3doc/reference/ |
| D | limitations.md | 46 x = tf.while_loop(..., loop_vars=(x,)
60 `loop_vars` because it is not affected by the `while` loop:
69 x = tf.while_loop(..., loop_vars=(x,) # y does not need to be a loop variable
220 loop_vars=(0,))
448 loop_vars=(tf.constant(1, dtype=tf.int32),)) # Error -- inconsistent dtypes: int32, float32
481 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])
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | while_v2_test.py | 1027 loop_vars=[
1047 loop_vars=[
1070 loop_vars=[
|
| D | control_flow_ops_py_test.py | 1499 body=body_fn, loop_vars=[1])
1516 body=body_fn, loop_vars=[0])
3374 loop_vars = [
3387 r = control_flow_ops.while_loop(c, b, loop_vars)
3402 loop_vars = [
3413 control_flow_ops.while_loop(c, b, loop_vars)
3484 loop_vars = [duration, variable, variable]
3486 cond=cond, body=body, loop_vars=loop_vars)
3505 loop_vars = [duration, c0]
3507 cond=cond, body=body, loop_vars=loop_vars)
|