Home
last modified time | relevance | path

Searched defs:extractOp (Results 1 – 20 of 20) sorted by relevance

/llvm-project/mlir/lib/Conversion/TensorToSPIRV/
H A DTensorToSPIRV.cpp44 matchAndRewrite(tensor::ExtractOp extractOp, OpAdaptor adaptor, in matchAndRewrite() argument
/llvm-project/mlir/lib/Dialect/Tensor/Transforms/
H A DSubsetInsertionOpInterfaceImpl.cpp62 auto extractOp = builder.create<tensor::ExtractSliceOp>( in buildSubsetExtraction() local
H A DExtractSliceFromReshapeUtils.cpp65 tensor::ExtractSliceOp extractOp) { in create()
H A DBufferizableOpInterfaceImpl.cpp436 auto extractOp = cast<tensor::ExtractOp>(op); bufferize() local
/llvm-project/mlir/lib/Dialect/Bufferization/Transforms/
H A DBufferDeallocationSimplification.cpp196 if (auto extractOp = in matchAndRewrite() local
414 auto extractOp = in matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/NVGPU/Transforms/
H A DCreateAsyncGroups.cpp72 if (auto extractOp = in getMaskOp() local
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DHoisting.cpp107 root->walk([&](vector::ExtractOp extractOp) { in hoistRedundantVectorBroadcasts() argument
[all...]
H A DVectorization.cpp732 tensor::ExtractOp extractOp = dyn_cast<tensor::ExtractOp>(op); tensorExtractVectorizationPrecondition() local
767 calculateGatherOffset(RewriterBase & rewriter,VectorizationState & state,tensor::ExtractOp extractOp,const IRMapping & bvm) calculateGatherOffset() argument
918 getTensorExtractMemoryAccessPattern(tensor::ExtractOp extractOp,LinalgOp & linalgOp) getTensorExtractMemoryAccessPattern() argument
1011 tensor::ExtractOp extractOp = dyn_cast<tensor::ExtractOp>(op); vectorizeTensorExtract() local
[all...]
/llvm-project/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp1353 foldExtractOpFromExtractChain(ExtractOp extractOp) foldExtractOpFromExtractChain() argument
1460 ExtractOp extractOp; global() member in __anon380139720f11::ExtractFromInsertTransposeChainState
1625 foldExtractFromBroadcast(ExtractOp extractOp) foldExtractFromBroadcast() argument
1684 foldExtractFromShapeCast(ExtractOp extractOp) foldExtractFromShapeCast() argument
1754 foldExtractFromExtractStrided(ExtractOp extractOp) foldExtractFromExtractStrided() argument
1806 foldExtractStridedOpFromInsertChain(ExtractOp extractOp) foldExtractStridedOpFromInsertChain() argument
1886 foldScalarExtractFromFromElements(ExtractOp extractOp) foldScalarExtractFromFromElements() argument
1949 matchAndRewrite(ExtractOp extractOp,PatternRewriter & rewriter) const matchAndRewrite() argument
1988 matchAndRewrite(ExtractOp extractOp,PatternRewriter & rewriter) const matchAndRewrite() argument
2013 matchAndRewrite(ExtractOp extractOp,PatternRewriter & rewriter) const matchAndRewrite() argument
2062 matchAndRewrite(ExtractOp extractOp,PatternRewriter & rewriter) const matchAndRewrite() argument
2124 foldExtractFromShapeCastToShapeCast(ExtractOp extractOp,PatternRewriter & rewriter) foldExtractFromShapeCastToShapeCast() argument
2149 foldExtractFromFromElements(ExtractOp extractOp,PatternRewriter & rewriter) foldExtractFromFromElements() argument
4700 auto extractOp = matchAndRewrite() local
[all...]
/llvm-project/mlir/lib/Dialect/Vector/Transforms/
H A DVectorTransferOpTransforms.cpp813 rewrite(vector::ExtractElementOp extractOp,PatternRewriter & rewriter) const rewrite() argument
857 rewrite(vector::ExtractOp extractOp,PatternRewriter & rewriter) const rewrite() argument
[all...]
H A DLowerVectorTranspose.cpp380 Value extractOp = in matchAndRewrite() local
H A DVectorDistribute.cpp1229 auto extractOp = operand->get().getDefiningOp<vector::ExtractOp>(); matchAndRewrite() local
1332 auto extractOp = operand->get().getDefiningOp<vector::ExtractElementOp>(); matchAndRewrite() local
[all...]
H A DVectorEmulateNarrowType.cpp52 if (auto extractOp = dyn_cast<vector::ExtractOp>(maskOp)) { getCompressedMaskOp() local
/llvm-project/mlir/lib/Conversion/VectorToSCF/
H A DVectorToSCF.cpp1210 if (auto extractOp = getExtractOp(xferOp)) getDataVector() local
1226 if (auto extractOp = getExtractOp(xferOp)) { getExtractionIndices() local
/llvm-project/mlir/lib/Dialect/MemRef/Transforms/
H A DExpandStridedMetadata.cpp898 matchAndRewrite(memref::ExtractAlignedPointerAsIndexOp extractOp,PatternRewriter & rewriter) const matchAndRewrite() argument
/llvm-project/mlir/lib/Conversion/MemRefToLLVM/
H A DMemRefToLLVM.cpp1590 matchAndRewrite(memref::ExtractAlignedPointerAsIndexOp extractOp,OpAdaptor adaptor,ConversionPatternRewriter & rewriter) const matchAndRewrite() argument
/llvm-project/flang/lib/Optimizer/Builder/
H A DPPCIntrinsicCall.cpp2386 auto extractOp{genVecExtract(resultType, args)}; in genVecSplat() local
/llvm-project/mlir/lib/Conversion/SPIRVToLLVM/
H A DSPIRVToLLVM.cpp1539 auto extractOp = rewriter.create<LLVM::ExtractElementOp>( matchAndRewrite() local
/llvm-project/mlir/lib/Conversion/VectorToLLVM/
H A DConvertVectorToLLVM.cpp1086 matchAndRewrite(vector::ExtractOp extractOp,OpAdaptor adaptor,ConversionPatternRewriter & rewriter) const matchAndRewrite() argument
/llvm-project/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp2450 foldExtractAfterInsertSlice(ExtractSliceOp extractOp) foldExtractAfterInsertSlice() argument
2597 auto extractOp = insertOp.getSource().getDefiningOp<ExtractSliceOp>(); foldInsertAfterExtractSlice() local