Searched refs:dfs_state_map (Results 1 – 1 of 1) sorted by relevance
98 absl::flat_hash_map<const HloInstruction*, DfsState> dfs_state_map; in TraverseAndPopulateCache() local101 InsertOrDie(&dfs_state_map, root, kDiscovered); in TraverseAndPopulateCache()110 switch (FindOrDie(dfs_state_map, instr)) { in TraverseAndPopulateCache()115 CHECK(!dfs_state_map.contains(operand) || in TraverseAndPopulateCache()116 dfs_state_map[operand] == kDiscovered); in TraverseAndPopulateCache()117 dfs_state_map[operand] = kDiscovered; in TraverseAndPopulateCache()120 dfs_state_map[instr] = kVisited; in TraverseAndPopulateCache()