Home
last modified time | relevance | path

Searched defs:loop_vars (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/python/autograph/operators/
Dcontrol_flow_test.py71 def set_state(loop_vars): argument
91 def set_state(loop_vars): argument
113 def set_state(loop_vars): argument
135 def set_state(loop_vars): argument
157 def set_state(loop_vars): argument
180 def set_state(loop_vars): argument
211 def set_state(loop_vars): argument
231 def set_state(loop_vars): argument
316 def set_state(loop_vars): argument
338 def set_state(loop_vars): argument
[all …]
Dcontrol_flow_deprecated_py2.py401 def while_body(iterate_index, *loop_vars):
412 def while_cond(iterate_index, *loop_vars):
465 def while_body(iterate_index, *loop_vars):
476 def while_cond(iterate_index, *loop_vars):
522 def while_body(iterate, *loop_vars):
531 def while_cond(iterate, *loop_vars):
577 def while_body_actual(opt_iterate, *loop_vars):
590 def while_body(has_next, *loop_vars):
613 def while_cond(has_next, *loop_vars):
Dcontrol_flow.py801 def reduce_body(loop_vars, iterate): argument
1072 def aug_test(*loop_vars):
1079 def aug_body(*loop_vars):
/external/tensorflow/tensorflow/python/ops/
Dwhile_v2_indexed_slices_rewriter.py31 def rewrite_grad_indexed_slices(grads, body_grad_graph, loop_vars, argument
112 forward_input, loop_vars): argument
250 def _update_indexed_slices_param(graph, loop_vars, init_slices, input_slices, argument
Dwhile_v2.py76 loop_vars, argument
456 def _build_while_op(loop_vars, cond_graph, body_graph, output_shapes, argument
1365 def _pack_sequence_as(structure_with_tas, loop_vars): argument
1381 def _tensor_array_to_flow(loop_vars): argument
Dcontrol_flow_ops.py2165 def _BuildLoop(self, pred, body, original_loop_vars, loop_vars, argument
2291 def BuildLoop(self, pred, body, loop_vars, shape_invariants, argument
2384 loop_vars, argument
2565 loop_vars, argument
/external/icu/icu4c/source/python/icutools/databuilder/
Dutils.py98 def format_repeated_request_command(request, cmd_template, loop_vars, common_vars): argument
/external/tensorflow/tensorflow/core/common_runtime/
Dexecutor_test.cc546 int loop_iters, int loop_vars, bool lower) { in BM_WhileLoopHelper()
662 const int loop_vars = state.range(1); in BM_LoweredWhileLoop() local
680 const int loop_vars = state.range(1); in BM_FunctionalWhileLoop() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dcholesky_expander.cc79 XlaBuilder* body_builder) -> StatusOr<std::vector<XlaOp>> { in CholeskyUnblocked()
/external/tensorflow/tensorflow/python/ops/risc/
Drisc_ops.py270 loop_vars, argument
/external/mesa3d/src/compiler/nir/
Dnir_loop_analyze.c68 nir_loop_variable *loop_vars; member