Searched refs:HloModuleGroupUtil (Results 1 – 2 of 2) sorted by relevance
41 std::vector<HloInstruction*> HloModuleGroupUtil::GlobalPredecessors( in GlobalPredecessors()118 std::vector<HloInstruction*> HloModuleGroupUtil::GlobalSuccessors( in GlobalSuccessors()195 std::vector<HloInstruction*> HloModuleGroupUtil::RootInstructions( in RootInstructions()210 string HloModuleGroupUtil::CycleToString(HloInstruction* init_instruction) { in CycleToString()243 Status HloModuleGroupUtil::VisitTopologicalOrder( in VisitTopologicalOrder()317 Status HloModuleGroupUtil::VerifyComputations( in VerifyComputations()348 HloModuleGroupUtil::ComputeReachability( in ComputeReachability()358 HloModuleGroupUtil::VisitStates visit_states; in ComputeReachability()370 void HloModuleGroupUtil::UpdateReachabilityThroughInstruction( in UpdateReachabilityThroughInstruction()
36 class HloModuleGroupUtil {38 explicit HloModuleGroupUtil(const HloModuleGroupMetadata& metadata) in HloModuleGroupUtil() function