Searched refs:HloCollectivePermuteInstruction (Results 1 – 3 of 3) sorted by relevance
601 HloCollectivePermuteInstruction::HloCollectivePermuteInstruction( in HloCollectivePermuteInstruction() function in xla::HloCollectivePermuteInstruction609 HloInstructionProto HloCollectivePermuteInstruction::ToProto() const { in ToProto()620 HloCollectivePermuteInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()631 bool HloCollectivePermuteInstruction::IdenticalSlowPath( in IdenticalSlowPath()636 static_cast<const HloCollectivePermuteInstruction&>(other); in IdenticalSlowPath()644 HloCollectivePermuteInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()647 return absl::make_unique<HloCollectivePermuteInstruction>( in CloneWithNewOperandsImpl()
370 class HloCollectivePermuteInstruction : public HloInstruction {372 explicit HloCollectivePermuteInstruction(
853 return absl::make_unique<HloCollectivePermuteInstruction>( in CreateCollectivePermute()3418 return Cast<HloCollectivePermuteInstruction>(this)->source_target_pairs(); in source_target_pairs()