Lines Matching defs:sizes
54 OpFoldResult offset, ArrayRef<OpFoldResult> sizes,
56 unsigned sourceRank = sizes.size();
57 assert(sizes.size() == strides.size() &&
58 "expected as many sizes as strides for a memref");
89 affine::makeComposedFoldedAffineApply(builder, loc, mulMap, sizes);
107 ArrayRef<OpFoldResult> sizes) {
108 SmallVector<OpFoldResult> strides(sizes.size());
109 if (!sizes.empty()) {
113 for (int index = sizes.size() - 1; index > 0; --index) {
116 ArrayRef<OpFoldResult>{strides[index], sizes[index]});
123 sizes, strides);
163 ArrayRef<OpFoldResult> sizes,
165 SmallVector<OpFoldResult> strides(sizes.size(), unit);
171 builder, loc, s0 * s1, {strides[r], sizes[r]});
178 ArrayRef<OpFoldResult> sizes) {
180 return computeSuffixProductIRBlockImpl(loc, builder, sizes, unit);