Searched refs:loop_switch_node (Results 1 – 1 of 1) sorted by relevance
299 Node* loop_switch_node; in GetLoopSwitchNode() local302 loop_switch_node = n; in GetLoopSwitchNode()307 TF_RET_CHECK(loop_switch_node->IsSwitch()) in GetLoopSwitchNode()309 return loop_switch_node; in GetLoopSwitchNode()314 Status GetOrCreateBeforeEachIterationNode(Graph* graph, Node* loop_switch_node, in GetOrCreateBeforeEachIterationNode() argument318 for (const auto out_edge : loop_switch_node->out_edges()) { in GetOrCreateBeforeEachIterationNode()329 TF_RETURN_IF_ERROR(GetNodeAttr(loop_switch_node->def(), "T", &dtype)); in GetOrCreateBeforeEachIterationNode()340 graph->AddEdge(loop_switch_node, 1, at_loop_iteration_node, 0); in GetOrCreateBeforeEachIterationNode()347 Status AddNoOpAfterLastIteration(Graph* graph, Node* loop_switch_node, in AddNoOpAfterLastIteration() argument351 for (const auto out_node : loop_switch_node->out_nodes()) { in AddNoOpAfterLastIteration()[all …]