Lines Matching defs:newDim
114 /// `newDim` at `iteratorTypes.size()` by:
115 /// 1. Appending `iteratorTypes[newDim]`, equal to `iteratorTypes[dim]`.
116 /// 2. Appending a `newDim` to the domain of every indexing map.
118 /// by potentially adding a `newDim` result to `map`.
152 int64_t newDim = iteratorTypes.size();
162 // Add the `newDim` to map whatever the case.
163 assert(map.getNumDims() == newDim && "num dims invariant violation");
164 map = map.shiftDims(1, newDim);
167 // If we can find one, we insert `AffineDimExpr(newDim)` to the map, which
168 // logically chunks dimension `dim` into `K * dim + newDim`, where the
182 // Add `newDim` to the results of the map.
183 map = map.insertResult(Builder(map.getContext()).getAffineDimExpr(newDim),