Searched refs:concat_pieces (Results 1 – 1 of 1) sorted by relevance
807 std::vector<HloInstruction*> concat_pieces; in ExchangeHalo() local847 concat_pieces.push_back(left_halo); in ExchangeHalo()850 concat_pieces.push_back(hlo); in ExchangeHalo()880 concat_pieces.push_back(right_halo); in ExchangeHalo()885 if (concat_pieces.size() > 1) { in ExchangeHalo()888 for (auto piece : concat_pieces) { in ExchangeHalo()893 HloInstruction::CreateConcatenate(concat_shape, concat_pieces, dim)); in ExchangeHalo()