Searched refs:CreateReduceScatter (Results 1 – 6 of 6) sorted by relevance
85 computation->AddInstruction(HloInstruction::CreateReduceScatter( in Run()
226 computation.AddInstruction(HloInstruction::CreateReduceScatter( in TryDecomposeAllReduce()
87 combined = computation.AddInstruction(HloInstruction::CreateReduceScatter( in CombineReduceScatters()
724 static std::unique_ptr<HloInstruction> CreateReduceScatter(
522 instruction = CreateReduceScatter( in CreateFromProto()1289 HloInstruction::CreateReduceScatter( in CreateReduceScatter() function in xla::HloInstruction
1499 return builder->AddInstruction(HloInstruction::CreateReduceScatter( in CreateInstruction()