Lines Matching refs:TransferWriteOp
31 using vector::TransferWriteOp;
254 if (xferOp.getOperationName().equals(TransferWriteOp::getOperationName())) { in isTensorOp()
448 struct Strategy<TransferWriteOp> {
456 static Value getBuffer(TransferWriteOp xferOp) { in getBuffer()
463 static void getBufferIndices(TransferWriteOp xferOp, in getBufferIndices()
479 static TransferWriteOp rewriteOp(OpBuilder &b, in rewriteOp()
481 TransferWriteOp xferOp, Value buffer, in rewriteOp()
495 auto newXferOp = b.create<vector::TransferWriteOp>( in rewriteOp()
506 static Value handleOutOfBoundsDim(OpBuilder &b, TransferWriteOp xferOp, in handleOutOfBoundsDim()
513 static void cleanup(PatternRewriter &rewriter, TransferWriteOp xferOp, in cleanup()
524 static Value initialLoopState(TransferWriteOp xferOp) { in initialLoopState()
618 : public VectorToSCFPattern<TransferWriteOp> {
619 using VectorToSCFPattern<TransferWriteOp>::VectorToSCFPattern;
621 LogicalResult matchAndRewrite(TransferWriteOp xferOp, in matchAndRewrite()
982 : public VectorToSCFPattern<TransferWriteOp> {
983 using VectorToSCFPattern<TransferWriteOp>::VectorToSCFPattern;
992 Value getDataVector(TransferWriteOp xferOp) const { in getDataVector()
999 vector::ExtractOp getExtractOp(TransferWriteOp xferOp) const { in getExtractOp()
1007 void getExtractionIndices(TransferWriteOp xferOp, in getExtractionIndices()
1017 LogicalResult matchAndRewrite(TransferWriteOp xferOp, in matchAndRewrite()
1056 auto newXferOp = b.create<vector::TransferWriteOp>( in matchAndRewrite()
1156 struct Strategy1d<TransferWriteOp> {
1158 TransferWriteOp xferOp, Value iv, in generateForLoopBody()
1174 static Value initialLoopState(OpBuilder &b, TransferWriteOp xferOp) { in initialLoopState()
1270 lowering_n_d::TransferOpConversion<TransferWriteOp>>( in populateVectorToSCFConversionPatterns()
1276 lowering_1_d::TransferOp1dConversion<TransferWriteOp>>( in populateVectorToSCFConversionPatterns()