Searched refs:input_concat_elements (Results 1 – 1 of 1) sorted by relevance
744 std::vector<HloInstruction*> input_concat_elements; in RewriteLoopWithConcatGroups() local745 input_concat_elements.reserve(group.elements.size()); in RewriteLoopWithConcatGroups()747 input_concat_elements.push_back(init_elements[param_gte->tuple_index()]); in RewriteLoopWithConcatGroups()750 group.CreateConcat(std::move(input_concat_elements), loop->parent()); in RewriteLoopWithConcatGroups()