Home
last modified time | relevance | path

Searched refs:enter_nodes_ (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dlower_while_op.cc160 std::vector<Node*> enter_nodes_; member in tensorflow::__anon29a522e20111::LowerWhileHelper
202 enter_nodes_.resize(num_loop_inputs_); in LowerWhileHelper()
253 enter_nodes_[edge->dst_input()] = enter_node; in CreateEnterNodes()
270 for (Node* n : enter_nodes_) { in CreateEnterNodes()
278 for (Node* enter_node : enter_nodes_) { in CreateMergeNodes()
298 cond_call_builder_.Input(NodeOut(enter_nodes_[i], 0)); in CreateCondFuncCallNode()
350 body_call_builder_.Input(NodeOut(enter_nodes_[i], 0)); in CreateBodyFuncCallNode()
387 TF_RETURN_IF_ERROR(enter_nodes_[i]->input_tensor(0, &resource_tensor)); in CreateExitNodes()
476 enter_nodes_[e->src_output()]->input_tensor(0, &resource)); in UpdateConsumers()
/external/tensorflow/tensorflow/core/graph/
Dwhile_context.cc27 enter_nodes_(std::move(enter_nodes)), in WhileContext()
32 const size_t num_loop_vars = enter_nodes_.size(); in WhileContext()
Dwhile_context.h43 const std::vector<Node*>& enter_nodes() const { return enter_nodes_; } in enter_nodes()
60 const std::vector<Node*> enter_nodes_; variable