Searched refs:exit_nodes (Results 1 – 6 of 6) 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
919 std::vector<Node*> exit_nodes, in AddWhileContext() argument926 WhileContext(frame_name, std::move(enter_nodes), std::move(exit_nodes), in AddWhileContext()
729 std::vector<Node*> exit_nodes,
551 absl::flat_hash_set<const NodeDef*> exit_nodes; in PruneGraphDef() local555 exit_nodes.insert(&node); in PruneGraphDef()571 for (const NodeDef* exit_node : exit_nodes) { in PruneGraphDef()
414 if (backprops.size() < while_ctx->exit_nodes().size()) return OkStatus(); in ProcessWhileLoop()421 for (Node* n : while_ctx->exit_nodes()) dy.push_back(backprops[n]); in ProcessWhileLoop()