Lines Matching defs:strides

26   SmallVector<ExprType> strides(sizes.size(), unit);
27   for (int64_t r = strides.size() - 2; r >= 0; --r)
28     strides[r] = strides[r + 1] * sizes[r + 1];
29   return strides;
56                                       ArrayRef<ExprType> strides,
58   int64_t rank = strides.size();
61     offsets[r] = divOp(linearIndex, strides[r]);
62     linearIndex = linearIndex % strides[r];
108                                        ArrayRef<int64_t> strides) {
109   assert(llvm::all_of(strides, [](int64_t s) { return s > 0; }) &&
110          "strides must be nonnegative");
111   return delinearizeImpl(linearIndex, strides,
188                                           ArrayRef<AffineExpr> strides) {
190       linearIndex, strides,
195                                           ArrayRef<int64_t> strides) {
197   return delinearize(linearIndex, getAffineConstantExprs(strides, ctx));
305                          ArrayRef<OpFoldResult> strides,
307   assert(strides.size() == indices.size());
308   auto sourceRank = static_cast<unsigned>(strides.size());
320     OpFoldResult origStride = strides[i];
335 mlir::computeLinearIndex(OpFoldResult sourceOffset, ArrayRef<int64_t> strides,
338       sourceOffset, getAsIndexOpFoldResult(sourceOffset.getContext(), strides),