Home
last modified time | relevance | path

Searched refs:ExtractOp (Results 1 – 25 of 34) sorted by relevance

12

/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorDropLeadUnitDim.cpp66 Value newSrcVector = rewriter.create<vector::ExtractOp>( in matchAndRewrite()
109 Value newSrcVector = rewriter.create<vector::ExtractOp>( in matchAndRewrite()
111 Value newDstVector = rewriter.create<vector::ExtractOp>( in matchAndRewrite()
158 newSrcVector = rewriter.create<vector::ExtractOp>( in matchAndRewrite()
161 Value newDstVector = rewriter.create<vector::ExtractOp>( in matchAndRewrite()
270 auto newVector = rewriter.create<vector::ExtractOp>( in matchAndRewrite()
387 ? rewriter.create<vector::ExtractOp>( in matchAndRewrite()
421 newOperands.push_back(rewriter.create<vector::ExtractOp>( in matchAndRewrite()
H A DVectorMultiDimReductionTransforms.cpp249 auto operand = rewriter.create<vector::ExtractOp>( in matchAndRewrite()
282 auto v = rewriter.create<vector::ExtractOp>( in matchAndRewrite()
284 auto acc = rewriter.create<vector::ExtractOp>( in matchAndRewrite()
334 rewriter.replaceOpWithNewOp<vector::ExtractOp>(multiReductionOp, reduced, in matchAndRewrite()
H A DVectorTransforms.cpp95 return rewriter.create<vector::ExtractOp>(loc, lowType, val, posAttr); in reshapeLoad()
452 rewriter.create<vector::ExtractOp>(loc, input, extractIdxs); in matchAndRewrite()
688 newLhs = rewriter.create<vector::ExtractOp>( in matchAndRewrite()
690 newRhs = rewriter.create<vector::ExtractOp>( in matchAndRewrite()
1491 Value a = builder.create<vector::ExtractOp>(loc, lhs, k); in outerProd()
1492 Value b = builder.create<vector::ExtractOp>(loc, rhs, k); in outerProd()
2159 extracted = rewriter.create<vector::ExtractOp>( in matchAndRewrite()
2248 : public OpRewritePattern<vector::ExtractOp> {
2251 LogicalResult matchAndRewrite(vector::ExtractOp extractOp, in matchAndRewrite()
2289 Value packedValue = rewriter.create<vector::ExtractOp>( in matchAndRewrite()
[all …]
H A DVectorInsertExtractStridedSliceRewritePatterns.cpp36 return rewriter.create<ExtractOp>(loc, vector, offset); in extractOne()
77 Value extracted = rewriter.create<ExtractOp>( in matchAndRewrite()
H A DVectorTransferPermutationMapRewritePatterns.cpp240 newRead = rewriter.create<tensor::ExtractOp>( in matchAndRewrite()
H A DVectorDistribute.cpp729 warpOp, [](Operation *op) { return isa<vector::ExtractOp>(op); }); in matchAndRewrite()
733 auto extractOp = operand->get().getDefiningOp<vector::ExtractOp>(); in matchAndRewrite()
742 Value newExtract = rewriter.create<vector::ExtractOp>( in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Conversion/TensorToSPIRV/
H A DTensorToSPIRV.cpp38 : public OpConversionPattern<tensor::ExtractOp> {
46 matchAndRewrite(tensor::ExtractOp extractOp, OpAdaptor adaptor, in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Conversion/TosaToSCF/
H A DTosaToSCF.cpp53 rewriter.create<tensor::ExtractOp>(yield.getLoc(), yield.getOperand(0)); in inlineWhileCase()
72 rewriter.create<tensor::ExtractOp>(op.getLoc(), op.getCond()); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Conversion/ShapeToStandard/
H A DShapeToStandard.cpp104 Value lesserRankOperandExtent = b.create<tensor::ExtractOp>( in getBroadcastedDim()
301 Value dimensionExtent = b.create<tensor::ExtractOp>( in matchAndRewrite()
352 rewriter.replaceOpWithNewOp<tensor::ExtractOp>(op, rewriter.getIndexType(), in matchAndRewrite()
410 Value extent = b.create<tensor::ExtractOp>(loc, adaptor.getShape(), iv); in matchAndRewrite()
508 b.create<tensor::ExtractOp>(loc, firstShape, iv); in matchAndRewrite()
509 Value rhsExtent = b.create<tensor::ExtractOp>(loc, shape, iv); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp1025 ExtractOp::Adaptor op(operands, attributes); in inferReturnTypes()
1051 LogicalResult vector::ExtractOp::verify() { in verify()
1082 ExtractOp currentOp = extractOp; in foldExtractOpFromExtractChain()
1085 while (ExtractOp nextOp = currentOp.getVector().getDefiningOp<ExtractOp>()) { in foldExtractOpFromExtractChain()
1109 ExtractFromInsertTransposeChainState(ExtractOp e);
1179 ExtractOp extractOp;
1200 ExtractOp e) in ExtractFromInsertTransposeChainState()
1500 op->setAttr(ExtractOp::getPositionAttrStrName(), in foldExtractStridedOpFromInsertChain()
1511 OpFoldResult ExtractOp::fold(ArrayRef<Attribute>) { in fold()
1536 LogicalResult matchAndRewrite(ExtractOp extractOp, in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp405 LogicalResult ExtractOp::verify() { in verify()
414 OpFoldResult ExtractOp::fold(ArrayRef<Attribute> operands) { in fold()
502 : public OpRewritePattern<tensor::ExtractOp> {
503 using OpRewritePattern<tensor::ExtractOp>::OpRewritePattern;
505 LogicalResult matchAndRewrite(tensor::ExtractOp extract, in matchAndRewrite()
514 auto newExtract = rewriter.create<tensor::ExtractOp>( in matchAndRewrite()
683 using OpRewritePattern<tensor::ExtractOp>::OpRewritePattern;
685 LogicalResult matchAndRewrite(tensor::ExtractOp extract, in matchAndRewrite()
713 using OpRewritePattern<tensor::ExtractOp>::OpRewritePattern;
715 LogicalResult matchAndRewrite(tensor::ExtractOp extract, in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DInlineScalarOperands.cpp76 Value extractedValue = rewriter.create<tensor::ExtractOp>( in matchAndRewrite()
H A DDetensorize.cpp152 return builder.create<tensor::ExtractOp>(loc, inputs[0], ValueRange{}); in DetensorizeTypeConverter()
/llvm-project-15.0.7/mlir/lib/Conversion/VectorToSCF/
H A DVectorToSCF.cpp808 auto newMask = b.create<vector::ExtractOp>(loc, xferOp.getMask(), indices); in maybeAssignMask()
999 vector::ExtractOp getExtractOp(TransferWriteOp xferOp) const { in getExtractOp()
1001 return dyn_cast<vector::ExtractOp>(op); in getExtractOp()
1002 return vector::ExtractOp(); in getExtractOp()
1054 b.create<vector::ExtractOp>(loc, vec, extractionIndices); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/
H A DResolveShapedTypeResultDims.cpp59 rewriter.replaceOpWithNewOp<tensor::ExtractOp>( in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalg.cpp1511 Value result = rewriter.create<tensor::ExtractOp>( in matchAndRewrite()
1547 Value y0x0 = rewriter.create<tensor::ExtractOp>( in matchAndRewrite()
1549 Value y0x1 = rewriter.create<tensor::ExtractOp>( in matchAndRewrite()
1551 Value y1x0 = rewriter.create<tensor::ExtractOp>( in matchAndRewrite()
1553 Value y1x1 = rewriter.create<tensor::ExtractOp>( in matchAndRewrite()
1758 auto extract = nestedBuilder.create<tensor::ExtractOp>( in matchAndRewrite()
1852 padConstant = rewriter.createOrFold<tensor::ExtractOp>( in matchAndRewrite()
1886 Value lowVal = rewriter.createOrFold<tensor::ExtractOp>( in matchAndRewrite()
1888 Value highVal = rewriter.createOrFold<tensor::ExtractOp>( in matchAndRewrite()
2085 Value extract = rewriter.create<tensor::ExtractOp>( in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/lib/Conversion/VectorToLLVM/
H A DConvertVectorToLLVM.cpp641 : public ConvertOpToLLVMPattern<vector::ExtractOp> {
643 using ConvertOpToLLVMPattern<vector::ExtractOp>::ConvertOpToLLVMPattern;
646 matchAndRewrite(vector::ExtractOp extractOp, OpAdaptor adaptor, in matchAndRewrite()
873 Value extrLHS = rewriter.create<ExtractOp>(loc, op.getLhs(), i); in matchAndRewrite()
874 Value extrRHS = rewriter.create<ExtractOp>(loc, op.getRhs(), i); in matchAndRewrite()
875 Value extrACC = rewriter.create<ExtractOp>(loc, op.getAcc(), i); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Conversion/VectorToSPIRV/
H A DVectorToSPIRV.cpp78 : public OpConversionPattern<vector::ExtractOp> {
82 matchAndRewrite(vector::ExtractOp extractOp, OpAdaptor adaptor, in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseTensorPasses.cpp140 linalg::YieldOp, tensor::ExtractOp>(); in runOnOperation()
H A DSparseTensorConversion.cpp246 Value val = builder.create<tensor::ExtractOp>(loc, tensor, ivs); in genIndexAndValueForDense()
316 Value val = builder.create<tensor::ExtractOp>(loc, indices, in genIndexAndValueForSparse()
321 return builder.create<tensor::ExtractOp>(loc, values, ivs[0]); in genIndexAndValueForSparse()
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/Transforms/
H A DBufferizableOpInterfaceImpl.cpp311 tensor::ExtractOp> {
329 auto extractOp = cast<tensor::ExtractOp>(op); in bufferize()
980 ExtractOp::attachInterface<ExtractOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
/llvm-project-15.0.7/mlir/lib/Conversion/MathToLibm/
H A DMathToLibm.cpp83 rewriter.create<vector::ExtractOp>(loc, input, positions)); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Rewrite/
H A DByteCode.cpp107 ExtractOp, enumerator
268 void generate(pdl_interp::ExtractOp op, ByteCodeWriter &writer);
741 pdl_interp::ExtractOp, pdl_interp::FinalizeOp, in generate()
871 void Generator::generate(pdl_interp::ExtractOp op, ByteCodeWriter &writer) { in generate()
874 .Case([](pdl::OperationType) { return OpCode::ExtractOp; }) in generate()
2067 case ExtractOp: in execute()
/llvm-project-15.0.7/mlir/lib/Dialect/X86Vector/Transforms/
H A DAVXTranspose.cpp299 vs.push_back(ib.create<vector::ExtractOp>(reshInput, i)); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/IR/
H A DBufferizationOps.cpp623 rewriter.replaceOpWithNewOp<tensor::ExtractOp>(load, toMemref.getTensor(), in matchAndRewrite()

12