Searched refs:loop_cond_node (Results 1 – 2 of 2) sorted by relevance
301 Status GetOrCreateBeforeEachIterationNode(const Node& loop_cond_node, in GetOrCreateBeforeEachIterationNode() argument304 for (auto n : loop_cond_node.out_nodes()) { in GetOrCreateBeforeEachIterationNode()349 Status AddNoOpAfterLastIteration(const Node& loop_cond_node, Graph* graph, in AddNoOpAfterLastIteration() argument362 for (auto switch_node : loop_cond_node.out_nodes()) { in AddNoOpAfterLastIteration()
572 TF_ASSIGN_OR_RETURN(const NodeDef* loop_cond_node, in PruneGraphDef()574 loop_cond_to_exit_nodes[loop_cond_node].insert(exit_node); in PruneGraphDef()