Lines Matching refs:getMask
121 if (!xferOp.getMask()) in generateMaskCheck()
129 return b.create<vector::ExtractElementOp>(loc, xferOp.getMask(), iv); in generateMaskCheck()
293 if (xferOp.getMask()) { in allocBuffers()
294 auto maskType = MemRefType::get({}, xferOp.getMask().getType()); in allocBuffers()
297 b.create<memref::StoreOp>(loc, xferOp.getMask(), maskBuffer); in allocBuffers()
323 assert(xferOp.getMask() && "Expected that transfer op has mask"); in getMaskBuffer()
324 auto loadOp = xferOp.getMask().template getDefiningOp<memref::LoadOp>(); in getMaskBuffer()
580 if (xferOp.getMask()) { in matchAndRewrite()
636 if (xferOp.getMask()) { in matchAndRewrite()
700 if (xferOp.getMask()) { in matchAndRewrite()
747 if (xferOp.getMask() && (xferOp.isBroadcastDim(0) || in matchAndRewrite()
791 if (!xferOp.getMask()) in maybeAssignMask()
797 newXferOp.getMaskMutable().assign(xferOp.getMask()); in maybeAssignMask()
808 auto newMask = b.create<vector::ExtractOp>(loc, xferOp.getMask(), indices); in maybeAssignMask()