Lines Matching defs:srcIdx

73     SmallVector<int64_t> srcIdx(srcRank - 1, 0);
83         incIdx(srcIdx, sourceVectorType, /*step=*/1);
88           rewriter.create<vector::ExtractOp>(loc, op.getSource(), srcIdx);
129     SmallVector<int64_t> srcIdx(srcRank, 0);
136         incIdx(srcIdx, sourceVectorType, /*step=*/extractSize);
141           loc, op.getSource(), /*offsets=*/srcIdx, /*sizes=*/extractSize,
185     SmallVector<int64_t> srcIdx(srcRank, 0);
191         incIdx(srcIdx, sourceVectorType);
198         assert(srcIdx.empty() && "Unexpected indices for 0-D vector");
203             rewriter.create<vector::ExtractOp>(loc, op.getSource(), srcIdx);
297     SmallVector<int64_t> srcIdx(srcRank, 0);
309               loc, op.getSource(), llvm::ArrayRef(srcIdx).drop_back());
317             loc, extractionVectorType, sourceSubVector, srcIdx.back());
346       if (srcIdx.back() + minExtractionSize >= minSourceTrailingSize) {
353       incIdx(srcIdx, sourceVectorType, /*step=*/minExtractionSize);