Lines Matching defs:newRetIndices

487     SmallVector<size_t> newRetIndices;
489         rewriter, warpOp, yieldValues, retTypes, newRetIndices);
499     newWriteOp.getVectorMutable().assign(newWarpOp.getResult(newRetIndices[0]));
547     SmallVector<size_t> newRetIndices;
552           TypeRange{targetType, maybeMaskType}, newRetIndices);
556           TypeRange{targetType}, newRetIndices);
562     newWriteOp.getVectorMutable().assign(newWarpOp.getResult(newRetIndices[0]));
564       newWriteOp.getMaskMutable().assign(newWarpOp.getResult(newRetIndices[1]));
622     SmallVector<size_t> newRetIndices;
624         rewriter, warpOp, yieldValues, retTypes, newRetIndices);
629       newOperands[i] = newWarpOp.getResult(newRetIndices[i]);
770     SmallVector<size_t> newRetIndices;
773         newRetIndices);
779       newIndices.push_back(newWarpOp.getResult(newRetIndices[i]));
797     Value newPadding = newWarpOp.getResult(newRetIndices[indices.size()]);
800         hasMask ? newWarpOp.getResult(newRetIndices[newRetIndices.size() - 1])
942     SmallVector<size_t> newRetIndices;
944         rewriter, warpOp, {broadcastSrc}, {broadcastSrcType}, newRetIndices);
947         loc, destVecType, newWarpOp->getResult(newRetIndices[0]));
984     SmallVector<size_t> newRetIndices;
987         newRetIndices);
991         newWarpOp->getResult(newRetIndices[0]));
1107       SmallVector<size_t> newRetIndices;
1110           {extractOp.getSourceVectorType()}, newRetIndices);
1112       Value distributedVec = newWarpOp->getResult(newRetIndices[0]);
1144     SmallVector<size_t> newRetIndices;
1147         newRetIndices);
1149     Value distributedVec = newWarpOp->getResult(newRetIndices[0]);
1208     SmallVector<size_t> newRetIndices;
1211         newRetIndices);
1213     Value distributedVec = newWarpOp->getResult(newRetIndices[0]);
1229                            ? (newWarpOp->getResult(newRetIndices[1]))
1310     SmallVector<size_t> newRetIndices;
1313         newRetIndices);
1315     Value distributedVec = newWarpOp->getResult(newRetIndices[0]);
1316     Value newSource = newWarpOp->getResult(newRetIndices[1]);
1323                 ? (newWarpOp->getResult(newRetIndices[2]))
1394       SmallVector<size_t> newRetIndices;
1398           newRetIndices);
1400       Value distributedSrc = newWarpOp->getResult(newRetIndices[0]);
1401       Value distributedDest = newWarpOp->getResult(newRetIndices[1]);
1440     SmallVector<size_t> newRetIndices;
1443         {distrSrcType, distrDestType}, newRetIndices);
1445     Value distributedSrc = newWarpOp->getResult(newRetIndices[0]);
1446     Value distributedDest = newWarpOp->getResult(newRetIndices[1]);
1576     SmallVector<size_t> newRetIndices;
1579         newRetIndices);
1611     for (auto [i, retIdx] : llvm::enumerate(newRetIndices)) {
1720     SmallVector<size_t> newRetIndices;
1722         rewriter, warpOp, yieldValues, retTypes, newRetIndices);
1726     Value laneValVec = newWarpOp.getResult(newRetIndices[0]);
1734           newWarpOp.getResult(newRetIndices[1]));