Home
last modified time | relevance | path

Searched defs:ConcatGroup (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_concat_code_motion.cc60 struct ConcatGroup { struct
61 ConcatGroup(std::vector<HloInstruction*> elements, int64_t concat_dim, in ConcatGroup() function
80 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;