Lines Matching defs:newXferOp

276 static void maybeApplyPassLabel(OpBuilder &b, OpTy newXferOp,
278   if (newXferOp.getVectorType().getRank() > targetRank)
279     newXferOp->setAttr(kPassLabel, b.getUnitAttr());
428     auto newXferOp = b.create<vector::TransferReadOp>(
433     maybeApplyPassLabel(b, newXferOp, options.targetRank);
435     b.create<memref::StoreOp>(loc, newXferOp.getVector(), buffer, storeIndices);
436     return newXferOp;
517     auto newXferOp = b.create<vector::TransferWriteOp>(
522     maybeApplyPassLabel(b, newXferOp, options.targetRank);
524     return newXferOp;
1189 static void maybeAssignMask(OpBuilder &b, OpTy xferOp, OpTy newXferOp,
1197     newXferOp.getMaskMutable().assign(xferOp.getMask());
1204     b.setInsertionPoint(newXferOp); // Insert load before newXfer.
1209     newXferOp.getMaskMutable().assign(newMask);
1336             auto newXferOp = b.create<vector::TransferReadOp>(
1340             maybeAssignMask(b, xferOp, newXferOp, i);
1341             return b.create<vector::InsertOp>(loc, newXferOp, vec,
1481             auto newXferOp = b.create<vector::TransferWriteOp>(
1486             maybeAssignMask(b, xferOp, newXferOp, i);
1488             return isTensorOp(xferOp) ? newXferOp->getResult(0) : Value();