Lines Matching defs:tileSliceIndex
30 /// rank 1: (indices[0] + (tileSliceIndex * tileSliceNumElts))
31 /// rank 2: (indices[0] + tileSliceIndex, indices[1])
33 Value tileSliceIndex,
39 auto tileSliceOffset = tileSliceIndex;
117 Value tileSliceIndex = forOp.getInductionVar();
120 memrefIndices, memrefRank, tileSliceIndex, numTileSlices, loc, rewriter);
122 tileSliceIndex, adjustedIndices, predicate,
208 [&](Value tileSliceIndex, ValueRange memrefIndices, Value predicate,
214 memrefIndices, tileSliceIndex, tileLoadOp.getLayout());
307 auto tileSliceIndex = forOp.getInductionVar();
312 loc, arith::CmpIPredicate::ult, tileSliceIndex, numRows);
325 tileSliceIndex, numTileSlices, loc, rewriter);
337 loc, tileType, loadSlice->getResult(0), currentTile, tileSliceIndex,
381 [&](Value tileSliceIndex, ValueRange memrefIndices, Value predicate) {
383 tileStoreOp, tileStoreOp.getValueToStore(), tileSliceIndex,