Searched refs:num_loop_vars (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/cc/ops/ |
| D | while_loop.cc | 132 const size_t num_loop_vars = inputs.size(); in CreateBody() local
133 if (outputs->size() != num_loop_vars) { in CreateBody()
135 "BuildWhileLoop: 'body' argument expected to return ", num_loop_vars, in CreateBody()
182 const size_t num_loop_vars = inputs.size(); in BuildWhileLoop() local
184 std::vector<Output> enter_outputs(num_loop_vars); in BuildWhileLoop()
185 for (int i = 0; i < num_loop_vars; ++i) { in BuildWhileLoop()
190 std::vector<Output> merge_outputs(num_loop_vars); in BuildWhileLoop()
191 for (int i = 0; i < num_loop_vars; ++i) { in BuildWhileLoop()
200 std::vector<Output> switch_trues(num_loop_vars); in BuildWhileLoop()
201 std::vector<Output> switch_falses(num_loop_vars); in BuildWhileLoop()
[all …]
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | while_context.cc | 32 const size_t num_loop_vars = enter_nodes_.size(); in WhileContext() local
33 DCHECK_EQ(exit_nodes_.size(), num_loop_vars); in WhileContext()
34 DCHECK_EQ(body_inputs_.size(), num_loop_vars); in WhileContext()
35 DCHECK_EQ(body_outputs_.size(), num_loop_vars); in WhileContext()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 2399 int num_loop_vars = params->ninputs; in TF_FinishWhileHelper() local
2421 [params, parent, num_loop_vars]( in TF_FinishWhileHelper()
2430 params->body_outputs, num_loop_vars, outputs)); in TF_FinishWhileHelper()
2443 scope, OutputsFromTFOutputs(parent_inputs, num_loop_vars), cond_fn, in TF_FinishWhileHelper()
2457 DCHECK_LE(loop_outputs.size(), num_loop_vars); in TF_FinishWhileHelper()
|