Searched refs:HloModuleGroupMetadata (Results 1 – 4 of 4) sorted by relevance
35 string HloModuleGroupMetadata::TrackedInstruction::ToString() const { in ToString()58 /* static */ StatusOr<std::unique_ptr<HloModuleGroupMetadata>>59 HloModuleGroupMetadata::Build(absl::Span<HloModule* const> modules) { in Build()60 auto metadata = absl::make_unique<HloModuleGroupMetadata>(modules); in Build()65 Status HloModuleGroupMetadata::Build() { in Build()144 Status HloModuleGroupMetadata::VerifyCompanionSets() const { in VerifyCompanionSets()181 bool HloModuleGroupMetadata::IsChannelInstruction( in IsChannelInstruction()198 bool HloModuleGroupMetadata::IsCompanionInstruction(HloInstruction* hlo) const { in IsCompanionInstruction()202 bool HloModuleGroupMetadata::InstructionCommunicates( in InstructionCommunicates()208 const HloModuleGroupMetadata::Channel& HloModuleGroupMetadata::GetChannel( in GetChannel()[all …]
63 class HloModuleGroupMetadata {113 explicit HloModuleGroupMetadata(absl::Span<HloModule* const> modules) in HloModuleGroupMetadata() function116 ~HloModuleGroupMetadata() = default;119 static StatusOr<std::unique_ptr<HloModuleGroupMetadata>> Build(
38 explicit HloModuleGroupUtil(const HloModuleGroupMetadata& metadata) in HloModuleGroupUtil()113 const HloModuleGroupMetadata& metadata_;
188 HloModuleGroupMetadata::Build(group.modules())); in TEST_F()