Searched refs:vect_dim (Results 1 – 2 of 2) sorted by relevance
140 static XlaOp RevectorizeInstr(XlaOp instr, int64_t dim, int64_t vect_dim, in RevectorizeInstr() argument146 CHECK_LE(size(vect_dim), vect_size); in RevectorizeInstr()147 CHECK_EQ(vect_size % size(vect_dim), 0); in RevectorizeInstr()149 int64_t split_factor = vect_size / size(vect_dim); in RevectorizeInstr()156 if (vect_dim > dim) { in RevectorizeInstr()157 vect_dim++; in RevectorizeInstr()161 instr = MoveDim(instr, dim + 1, vect_dim); in RevectorizeInstr()165 if (vect_dim > dim) { in RevectorizeInstr()166 vect_dim--; in RevectorizeInstr()170 return Collapse(instr, {vect_dim, vect_dim + 1}); in RevectorizeInstr()[all …]
465 DimensionHandle vect_dim = c->Dim( in CheckFormatConstraintsOnShape() local467 int64_t vect_dim_val = c->Value(vect_dim); in CheckFormatConstraintsOnShape()