Searched defs:ConcatGroup (Results 1 – 1 of 1) sorted by relevance
60 struct ConcatGroup { struct61 ConcatGroup(std::vector<HloInstruction*> elements, int64_t concat_dim, in ConcatGroup() function80 Shape GetConcatShape() const { in GetConcatShape()106 HloInstruction* CreateSlice(HloInstruction* full_data, int64_t element_index, in CreateSlice()131 HloInstruction* CreateConcat(std::vector<HloInstruction*> input_elements, in CreateConcat()157 std::vector<HloInstruction*> elements;158 std::vector<int64> element_sizes;159 std::vector<int64> element_offsets;160 int64 concat_dim;162 bool inserted_concat_dim;