Searched refs:exit_nodes (Results 1 – 5 of 5) sorted by relevance
22 std::vector<Node*> exit_nodes, in WhileContext() argument28 exit_nodes_(std::move(exit_nodes)), in WhileContext()
38 std::vector<Node*> exit_nodes, OutputTensor cond_output,44 const std::vector<Node*>& exit_nodes() const { return exit_nodes_; } in exit_nodes() function
794 std::vector<Node*> exit_nodes, in AddWhileContext() argument801 WhileContext(frame_name, std::move(enter_nodes), std::move(exit_nodes), in AddWhileContext()
645 std::vector<Node*> exit_nodes,
413 if (backprops.size() < while_ctx->exit_nodes().size()) return Status::OK(); in ProcessWhileLoop()420 for (Node* n : while_ctx->exit_nodes()) dy.push_back(backprops[n]); in ProcessWhileLoop()