Searched refs:HloCollectiveInstruction (Results 1 – 3 of 3) sorted by relevance
476 HloCollectiveInstruction::HloCollectiveInstruction( in HloCollectiveInstruction() function in xla::HloCollectiveInstruction486 HloInstructionProto HloCollectiveInstruction::ToProto() const { in ToProto()493 std::vector<string> HloCollectiveInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()506 bool HloCollectiveInstruction::IdenticalSlowPath( in IdenticalSlowPath()511 static_cast<const HloCollectiveInstruction&>(other); in IdenticalSlowPath()523 : HloCollectiveInstruction(HloOpcode::kAllReduce, shape, operands, in HloAllReduceInstruction()536 HloInstructionProto proto = HloCollectiveInstruction::ToProto(); in ToProto()557 HloCollectiveInstruction::ExtraAttributesToStringImpl(options); in ExtraAttributesToStringImpl()572 return HloCollectiveInstruction::IdenticalSlowPath(other, eq_computations) && in IdenticalSlowPath()590 : HloCollectiveInstruction(HloOpcode::kAllToAll, shape, operands, in HloAllToAllInstruction()
288 class HloCollectiveInstruction : public HloInstruction {295 explicit HloCollectiveInstruction(312 class HloAllReduceInstruction : public HloCollectiveInstruction {357 class HloAllToAllInstruction : public HloCollectiveInstruction {
3413 return Cast<HloCollectiveInstruction>(this)->replica_groups(); in replica_groups()