Lines Matching defs:linearIndex
48 ExprType linearIndex = zero;
50 linearIndex = linearIndex + offsets[idx] * basis[idx];
51 return linearIndex;
55 SmallVector<ExprType> delinearizeImpl(ExprType linearIndex,
61 offsets[r] = divOp(linearIndex, strides[r]);
62 linearIndex = linearIndex % strides[r];
107 SmallVector<int64_t> mlir::delinearize(int64_t linearIndex,
111 return delinearizeImpl(linearIndex, strides,
187 SmallVector<AffineExpr> mlir::delinearize(AffineExpr linearIndex,
190 linearIndex, strides,
194 SmallVector<AffineExpr> mlir::delinearize(AffineExpr linearIndex,
196 MLIRContext *ctx = linearIndex.getContext();
197 return delinearize(linearIndex, getAffineConstantExprs(strides, ctx));
379 int64_t linearIndex) const {
381 delinearize(linearIndex, sliceStrides), inverseLoopOrder);
387 AffineExpr linearIndex) const {
388 MLIRContext *ctx = linearIndex.getContext();
390 delinearize(linearIndex, sliceStrides), inverseLoopOrder);