Searched refs:HloReduceScatterInstruction (Results 1 – 10 of 10) sorted by relevance
82 const auto* rs = Cast<HloReduceScatterInstruction>(to_combine.front()); in CombineReduceScatters()143 auto* rs = DynCast<HloReduceScatterInstruction>(instruction); in Run()
47 auto *rs = DynCast<HloReduceScatterInstruction>(instruction); in Run()
896 HloReduceScatterInstruction::HloReduceScatterInstruction( in HloReduceScatterInstruction() function in xla::HloReduceScatterInstruction908 HloReduceScatterInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()916 HloInstructionProto HloReduceScatterInstruction::ToProto() const { in ToProto()922 bool HloReduceScatterInstruction::IdenticalSlowPathIgnoringChannelIdValues( in IdenticalSlowPathIgnoringChannelIdValues()927 static_cast<const HloReduceScatterInstruction&>(other); in IdenticalSlowPathIgnoringChannelIdValues()934 HloReduceScatterInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()937 return std::make_unique<HloReduceScatterInstruction>( in CloneWithNewOperandsImpl()
677 class HloReduceScatterInstruction : public HloAllReduceInstructionBase {679 explicit HloReduceScatterInstruction(
427 auto ars = Cast<HloReduceScatterInstruction>(hlo); in HandleReduceScatter()
1295 return std::make_unique<HloReduceScatterInstruction>( in CreateReduceScatter()
97 const auto *rs = Cast<HloReduceScatterInstruction>( in TEST_F()203 const auto *rs = Cast<HloReduceScatterInstruction>( in TEST_F()
376 auto ars = Cast<HloReduceScatterInstruction>(instruction); in AddBackendConstraints()
1186 auto reduce_scatter = Cast<HloReduceScatterInstruction>(instruction); in ImportInstructionImpl()
1236 auto* ars = xla::Cast<xla::HloReduceScatterInstruction>(instr); in EmitReduceScatterOp()