| /external/tensorflow/tensorflow/python/ops/ |
| while_v2_indexed_slices_rewriter.py | 31 def rewrite_grad_indexed_slices(grads, body_grad_graph, loop_vars,
43 into an input IndexedSlices. It also returns new `loop_vars` to reflect the
49 loop_vars: list of Tensors. The inputs to body_grad_graph.
54 The new loop_vars to pass to body_grad_graph.
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):
118 loop_vars: list of Tensors. The inputs to body_grad_graph.
121 The new loop_vars to pass to body_grad_graph
[all...] |
| while_v2.py | 58 loop_vars,
65 # Keep the original loop_vars around to know which args were TensorArrays.
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)
[all...] |
| gradients_impl.py | 379 loop_vars = [
389 loop_vars
|
| control_flow_ops.py | [all...] |
| rnn.py | [all...] |
| /external/tensorflow/tensorflow/python/keras/utils/ |
| 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/python/grappler/ |
| constant_folding_test.py | 59 loop_vars=[0, init_y],
|
| tf_optimizer_test.py | 105 loop_vars = [init_buf, start]
110 buf, _ = control_flow_ops.while_loop(_Cond, _Body, loop_vars, shape_inv)
|
| hierarchical_controller.py | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| cholesky_expander.cc | 69 [&](XlaOp i, absl::Span<const XlaOp> loop_vars,
91 auto body_a = loop_vars[0];
92 auto body_l = loop_vars[1];
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| random_ops.cc | 159 absl::Span<const xla::XlaOp> loop_vars,
162 auto swaps = loop_vars[0];
163 auto indices = loop_vars[1];
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| resample.py | 59 cond, body, loop_vars=[init_output_array, 0])
|
| /external/mesa3d/src/compiler/nir/ |
| nir_loop_analyze.c | 69 nir_loop_variable *loop_vars; member in struct:__anon33132
71 /* A list of the loop_vars to analyze */
81 return &(state->loop_vars[value->index]);
785 state->loop_vars = rzalloc_array(mem_ctx, nir_loop_variable,
|
| /external/tensorflow/tensorflow/contrib/constrained_optimization/python/ |
| swap_regret_optimizer.py | 121 loop_vars=(iteration, eigenvector, old_eigenvector),
193 loop_vars=(iteration, matrix, inactive, old_inactive),
|
| external_regret_optimizer.py | 120 loop_vars=(iteration, multipliers, inactive, old_inactive),
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/ |
| masked_autoregressive.py | 288 loop_vars=(0, y0),
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| decoder.py | 445 loop_vars=(
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| control_flow_ops_py_test.py | [all...] |
| tensor_array_ops_test.py | [all...] |
| cond_v2_test.py | 826 loop_vars=(constant_op.constant(0), output))
855 loop_vars=(constant_op.constant(0), output))
[all...] |
| /external/tensorflow/tensorflow/contrib/graph_editor/tests/ |
| transform_test.py | 252 loop_vars=[index_start, sum_start])
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| model.py | [all...] |
| /external/v8/src/builtins/ |
| builtins-number-gen.cc | 374 Variable* loop_vars[2] = {&var_left, &var_right}; local
375 Label loop(this, 2, loop_vars),
[all...] |
| /external/v8/src/interpreter/ |
| interpreter-generator.cc | 1143 Variable* loop_vars[] = {&var_value, &var_feedback}; local
[all...] |
| /external/tensorflow/tensorflow/contrib/compiler/ |
| xla_test.py | 157 cond=lambda i: math_ops.less(i, 10), body=while_body, loop_vars=[i])
|
