Home
last modified time | relevance | path

Searched defs:loop_vars (Results 1 – 12 of 12) 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.py402 def while_body(iterate_index, *loop_vars):
413 def while_cond(iterate_index, *loop_vars):
466 def while_body(iterate_index, *loop_vars):
477 def while_cond(iterate_index, *loop_vars):
523 def while_body(iterate, *loop_vars):
532 def while_cond(iterate, *loop_vars):
578 def while_body_actual(opt_iterate, *loop_vars):
591 def while_body(has_next, *loop_vars):
614 def while_cond(has_next, *loop_vars):
Dcontrol_flow.py800 def reduce_body(loop_vars, iterate): argument
1070 def aug_test(*loop_vars):
1077 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
110 forward_input, loop_vars): argument
248 def _update_indexed_slices_param(graph, loop_vars, init_slices, input_slices, argument
Dwhile_v2.py62 loop_vars, argument
424 def _build_while_op(loop_vars, cond_graph, body_graph, output_shapes, argument
1317 def _pack_sequence_as(structure_with_tas, loop_vars): argument
1333 def _tensor_array_to_flow(loop_vars): argument
Dcontrol_flow_ops.py2150 def _BuildLoop(self, pred, body, original_loop_vars, loop_vars, argument
2275 def BuildLoop(self, pred, body, loop_vars, shape_invariants, argument
2367 loop_vars, argument
2548 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/compiler/tf2xla/kernels/
Drandom_ops.cc172 -> xla::StatusOr<std::vector<xla::XlaOp>> { in Compile()
/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