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()132 HloInstruction* CreateConcat(std::vector<HloInstruction*> input_elements, in CreateConcat()158 std::vector<HloInstruction*> elements;159 std::vector<int64_t> element_sizes;160 std::vector<int64_t> element_offsets;161 int64_t concat_dim;163 bool inserted_concat_dim;