| /external/tensorflow/tensorflow/python/ops/ |
| D | while_v2.py | 58 loop_vars, argument
66 orig_loop_vars = loop_vars
73 loop_vars = list(_tensor_array_to_flow(orig_loop_vars))
74 loop_vars = nest.map_structure(
75 ops.internal_convert_to_tensor_or_indexed_slices, loop_vars)
79 shape_invariants = nest.map_structure(lambda t: t.shape, loop_vars)
96 loop_vars = [loop_counter, maximum_iterations_loop_var] + loop_vars
126 signature=_build_signature(loop_vars, shape_invariants),
170 signature=_build_signature(loop_vars, shape_invariants),
178 loop_vars = loop_vars + body_graph.external_captures
[all …]
|
| 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 | control_flow_ops.py | 2850 def _BuildLoop(self, pred, body, original_loop_vars, loop_vars, argument
2857 self._InitializeValues(loop_vars)
2858 real_vars = loop_vars
2860 real_vars = [self._outer_context.AddValue(x) for x in loop_vars]
2975 def BuildLoop(self, pred, body, loop_vars, shape_invariants, argument
2980 original_loop_vars = loop_vars
2982 loop_vars = nest.map_structure(
2984 nest.flatten(loop_vars, expand_composites=False),
2986 loop_vars = ops.convert_n_to_tensor_or_composite(loop_vars)
2989 _get_shape_invariant, loop_vars, expand_composites=False)
[all …]
|
| D | gradients_impl.py | 379 loop_vars = [
389 loop_vars
|
| D | rnn.py | 876 loop_vars=(time, output_ta, state),
1206 condition, body, loop_vars=[
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | metrics_utils.py | 342 loop_vars = {
351 loop_vars[ConfusionMatrix.FALSE_NEGATIVES] = (label_is_pos, pred_is_neg)
355 loop_vars[ConfusionMatrix.FALSE_POSITIVES] = (label_is_neg, pred_is_pos)
357 loop_vars[ConfusionMatrix.TRUE_NEGATIVES] = (label_is_neg, pred_is_neg)
359 for matrix_cond, (label, pred) in loop_vars.items():
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | cholesky_expander.cc | 69 [&](XlaOp i, absl::Span<const XlaOp> loop_vars, in CholeskyUnblocked()
91 auto body_a = loop_vars[0]; in CholeskyUnblocked()
92 auto body_l = loop_vars[1]; in CholeskyUnblocked()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | random_ops.cc | 159 absl::Span<const xla::XlaOp> loop_vars, in Compile()
162 auto swaps = loop_vars[0]; in Compile()
163 auto indices = loop_vars[1]; in Compile()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | tf_optimizer_test.py | 105 loop_vars = [init_buf, start]
110 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 | 934 loop_vars = [
938 loop_outputs = control_flow_ops.while_loop(condition, body, loop_vars)
|
| /external/tensorflow/tensorflow/contrib/constrained_optimization/python/ |
| D | swap_regret_optimizer.py | 121 loop_vars=(iteration, eigenvector, old_eigenvector),
193 loop_vars=(iteration, matrix, inactive, old_inactive),
|
| D | external_regret_optimizer.py | 120 loop_vars=(iteration, multipliers, inactive, old_inactive),
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | resample.py | 59 cond, body, loop_vars=[init_output_array, 0])
|
| /external/mesa3d/src/compiler/nir/ |
| D | nir_loop_analyze.c | 69 nir_loop_variable *loop_vars; member
81 return &(state->loop_vars[value->index]); in get_loop_var()
785 state->loop_vars = rzalloc_array(mem_ctx, nir_loop_variable, in initialize_loop_info_state()
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| D | decoder.py | 445 loop_vars=(
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/ |
| D | masked_autoregressive.py | 288 loop_vars=(0, y0),
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | control_flow_ops_py_test.py | 1308 body=body_fn, loop_vars=[1])
1325 body=body_fn, loop_vars=[0])
3013 loop_vars = [
3026 r = control_flow_ops.while_loop(c, b, loop_vars)
3041 loop_vars = [
3052 control_flow_ops.while_loop(c, b, loop_vars)
3124 loop_vars = [duration, variable, variable]
3126 cond=cond, body=body, loop_vars=loop_vars)
3145 loop_vars = [duration, c0]
3147 cond=cond, body=body, loop_vars=loop_vars)
|
| D | cond_v2_test.py | 826 loop_vars=(constant_op.constant(0), output))
855 loop_vars=(constant_op.constant(0), output))
|
| D | tensor_array_ops_test.py | 945 loop_vars=(time_0, ta, state0),
1544 lambda i, ta: i < 2, _body, loop_vars=[0, ta])
1576 lambda i, ta: i < 2, _body, loop_vars=[0, ta])
|
| /external/tensorflow/tensorflow/contrib/graph_editor/tests/ |
| D | transform_test.py | 252 loop_vars=[index_start, sum_start])
|
| /external/v8/src/builtins/ |
| D | builtins-number-gen.cc | 374 Variable* loop_vars[2] = {&var_left, &var_right}; in TF_BUILTIN() local
375 Label loop(this, 2, loop_vars), in TF_BUILTIN()
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| D | model.py | 802 loop_vars=initial_loop_arguments)
|
| /external/tensorflow/tensorflow/contrib/compiler/ |
| D | xla_test.py | 157 cond=lambda i: math_ops.less(i, 10), body=while_body, loop_vars=[i])
|
| /external/v8/src/interpreter/ |
| D | interpreter-generator.cc | 1143 Variable* loop_vars[] = {&var_value, &var_feedback}; in UnaryOpWithFeedback() local
1144 Label start(this, arraysize(loop_vars), loop_vars), end(this); in UnaryOpWithFeedback()
|