Searched refs:loop_switch_node (Results 1 – 1 of 1) sorted by relevance
303 Node* loop_switch_node = nullptr; in GetOrCreateBeforeEachIterationNode() local306 loop_switch_node = n; in GetOrCreateBeforeEachIterationNode()310 TF_RET_CHECK(loop_switch_node != nullptr) in GetOrCreateBeforeEachIterationNode()315 for (const auto out_edge : loop_switch_node->out_edges()) { in GetOrCreateBeforeEachIterationNode()326 TF_RETURN_IF_ERROR(GetNodeAttr(loop_switch_node->def(), "T", &dtype)); in GetOrCreateBeforeEachIterationNode()337 graph->AddEdge(loop_switch_node, 1, at_loop_iteration_node, 0); in GetOrCreateBeforeEachIterationNode()