Searched refs:visited_nodes (Results 1 – 2 of 2) sorted by relevance
155 std::unordered_set<const NodeDef*> visited_nodes; in GetOpGroupsToRecompute() local160 if (visited_nodes.count(recompute_node) > 0) { in GetOpGroupsToRecompute()173 visited_nodes.insert(unpruned_recompute_nodes.begin(), in GetOpGroupsToRecompute()
580 const std::vector<const HloInstruction*>& visited_nodes() { in visited_nodes() function in xla::__anon9b186d4a0111::NodeCollectorAndPostProcessor621 EXPECT_EQ(visitor.visited_nodes(), visitor.post_processed_nodes()); in TEST_F()623 EXPECT_TRUE(Distinct(visitor.visited_nodes())); in TEST_F()