Home
last modified time | relevance | path

Searched defs:sliceOp (Results 1 – 16 of 16) sorted by relevance

/llvm-project/mlir/lib/Dialect/Tensor/Transforms/
H A DSwapExtractSliceWithProducerPatterns.cpp24 OpBuilder &builder, tensor::ExtractSliceOp sliceOp, OpResult producer) { in replaceExtractSliceWithTiledProducer()
45 OpBuilder &builder, OffsetSizeAndStrideOpInterface sliceOp, in replaceInsertSliceWithTiledConsumer()
H A DExtractSliceFromReshapeUtils.cpp178 auto sliceOp = rewriter.create<tensor::ExtractSliceOp>( in simplifyCollapseShapeWithRankReducingExtractSlice() local
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DFusion.cpp175 if (auto sliceOp = dyn_cast<tensor::ExtractSliceOp>(shapeProducingOp)) in getRangeFromOperandShape() local
212 if (auto sliceOp = tensor.getDefiningOp<tensor::ExtractSliceOp>()) { in getProducerOfTensor() local
252 auto sliceOp = inputTensor.getDefiningOp<tensor::ExtractSliceOp>(); in fuseProducerOfTensor() local
H A DHoisting.cpp232 for (auto *sliceOp : llvm::reverse(forwardSlice)) { hoistRedundantVectorTransfers() local
H A DHoistPadding.cpp238 tensor::ExtractSliceOp sliceOp; member
573 if (auto sliceOp = dyn_cast<tensor::ExtractSliceOp>(op)) { buildPackingLoopNestImpl() local
[all...]
H A DTransforms.cpp1000 matchAndRewrite(tensor::ExtractSliceOp sliceOp,PatternRewriter & rewriter) const matchAndRewrite() argument
/llvm-project/flang/lib/Optimizer/CodeGen/
H A DPreCGRewrite.cpp136 if (auto sliceOp = in rewriteDynamicShape() local
193 if (auto sliceOp = in matchAndRewrite() local
251 if (auto sliceOp = matchAndRewrite() local
[all...]
/llvm-project/mlir/lib/Dialect/Affine/Transforms/
H A DPipelineDataTransfer.cpp337 for (auto sliceOp : sliceOps) { in runOnAffineForOp() local
/llvm-project/mlir/lib/Dialect/SCF/Transforms/
H A DTileUsingInterface.cpp954 yieldReplacementForFusedProducer(RewriterBase & rewriter,tensor::ExtractSliceOp sliceOp,scf::SCFFuseProducerOfSliceResult fusedProducerInfo,MutableArrayRef<LoopLikeOpInterface> loops,ArrayRef<unsigned> yieldResultNumber) yieldReplacementForFusedProducer() argument
1128 if (auto sliceOp = operand.getDefiningOp<tensor::ExtractSliceOp>()) tileConsumerAndFuseProducersUsingSCF() local
1319 getUntiledConsumerFromSlice(Operation * sliceOp) getUntiledConsumerFromSlice() argument
1485 if (auto sliceOp = tileAndFuseConsumerOfSlice() local
[all...]
/llvm-project/mlir/lib/Dialect/Linalg/Utils/
H A DUtils.cpp195 auto sliceOp = source.getDefiningOp<tensor::ExtractSliceOp>(); in makeComposedPadHighOp() local
573 auto *sliceOp = TypeSwitch<ShapedType, Operation *>(shapedType) materializeTiledShape() local
762 if (auto sliceOp = outputTensor.getDefiningOp<tensor::ExtractSliceOp>()) { insertSlicesBack() local
[all...]
/llvm-project/mlir/lib/Conversion/TosaToTensor/
H A DTosaToTensor.cpp264 matchAndRewrite(tosa::SliceOp sliceOp, OpAdaptor adaptor, in matchAndRewrite() argument
[all...]
/llvm-project/flang/lib/Optimizer/Transforms/
H A DArrayValueCopy.cpp479 auto sliceOp = mlir::dyn_cast_or_null<SliceOp>(val.getDefiningOp()); in mutuallyExclusiveSliceRange() local
862 if (auto sliceOp = in getAdjustedExtents() local
997 mlir::Value sliceOp, ArrayLoadOp arrLoad) { in genArrayCopy()
/llvm-project/flang/lib/Optimizer/Dialect/
H A DFIROps.cpp410 if (auto sliceOp = getSlice()) { verify() local
481 if (auto sliceOp = matchAndRewrite() local
848 if (auto sliceOp = getSlice()) { verify() local
871 if (auto sliceOp = verify() local
2815 if (auto sliceOp = sliceVal.getDefiningOp()) { verify() local
[all...]
/llvm-project/mlir/lib/Dialect/Linalg/TransformOps/
H A DLinalgTransformOps.cpp702 auto sliceOp = dyn_cast<tensor::ExtractSliceOp>(user); tileAndFuseFirstExtractUse() local
807 auto sliceOp = dyn_cast<tensor::ExtractSliceOp>(user); tileAndFuseFirstExtractUseThroughContainingOpBlockArgument() local
/llvm-project/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp783 if (auto sliceOp = dyn_cast_or_null<tensor::ExtractSliceOp>(definingOp)) { fold() local
2239 matchAndRewrite(ExtractSliceOp sliceOp,PatternRewriter & rewriter) const matchAndRewrite() argument
[all...]
/llvm-project/flang/lib/Optimizer/Builder/
H A DFIRBuilder.cpp1215 auto sliceOp = mlir::dyn_cast_or_null<fir::SliceOp>(slice.getDefiningOp()); arraySectionElementToExtendedValue() local