Searched refs:reachability_map (Results 1 – 4 of 4) sorted by relevance
126 HloReachabilityMap& reachability_map(const HloComputation* computation) { in reachability_map() function129 const HloReachabilityMap& reachability_map( in reachability_map() function
108 HloInstruction* instruction, HloReachabilityMap* reachability_map);
366 HloInstruction* instruction, HloReachabilityMap* reachability_map) { in UpdateReachabilityThroughInstruction() argument373 if (reachability_map->SetReachabilityToUnion(GlobalPredecessors(item), in UpdateReachabilityThroughInstruction()
5491 std::unique_ptr<HloReachabilityMap> reachability_map = in DivideOutputInstructionsIntoGroups() local5505 if (reachability_map->IsReachable(instr, reduce)) { in DivideOutputInstructionsIntoGroups()