| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Padding.cpp | 195 if (auto padOp = paddedOperand->getDefiningOp<tensor::PadOp>()) in rewriteAsPaddedOp() local
288 auto padOp = opOperand.get().getDefiningOp<tensor::PadOp>(); in padAndHoistLinalgOp() local
|
| H A D | HoistPadding.cpp | 74 getAtMostNEnclosingLoops(tensor::PadOp padOp, int nLevels, in getAtMostNEnclosingLoops() argument
93 getEnclosingLoopsUntil(tensor::PadOp padOp, scf::ForOp untilLoop, in getEnclosingLoopsUntil() argument
111 static void computeBackwardSlice(tensor::PadOp padOp, in computeBackwardSlice() argument
246 HoistPaddingAnalysis(tensor::PadOp padOp,int numLoops) HoistPaddingAnalysis() argument
264 HoistPaddingAnalysis(tensor::PadOp padOp,scf::ForOp outermostEnclosingForOp) HoistPaddingAnalysis() argument
[all...] |
| H A D | ConvertToDestinationStyle.cpp | 96 Location loc, PadOp padOp, in movePaddingToFillOrGenericOp() argument
224 bufferizeToAllocation(RewriterBase & rewriter,const linalg::BufferizeToAllocationOptions & options,PadOp padOp,Attribute memorySpace,Operation * insertionPoint) bufferizeToAllocation() argument
425 rewriteInDestinationPassingStyle(RewriterBase & rewriter,tensor::PadOp padOp) rewriteInDestinationPassingStyle() argument
477 if (auto padOp = dyn_cast<tensor::PadOp>(op)) bufferizeToAllocation() local
[all...] |
| H A D | Transforms.cpp | 275 auto padOp = lowerPack() local
928 createFillOrGenerateOp(RewriterBase & rewriter,tensor::PadOp padOp,Value dest,const SmallVector<Value> & dynSizes) const createFillOrGenerateOp() argument
944 matchAndRewrite(tensor::PadOp padOp,PatternRewriter & rewriter) const matchAndRewrite() argument
1004 auto padOp = sliceOp.getSource().getDefiningOp<tensor::PadOp>(); matchAndRewrite() local
[all...] |
| H A D | Vectorization.cpp | 1695 vectorizeAsTensorPadOp(RewriterBase & rewriter,tensor::PadOp padOp,ArrayRef<int64_t> inputVectorSizes,SmallVectorImpl<Value> & newResults) vectorizeAsTensorPadOp() argument
1910 vectorizePadOpPrecondition(tensor::PadOp padOp,ArrayRef<int64_t> inputVectorSizes) vectorizePadOpPrecondition() argument
1970 __anon729377362002(auto padOp) vectorizeOpPrecondition() argument
2066 __anon729377362502(auto padOp) vectorize() argument
[all...] |
| H A D | DataLayoutPropagation.cpp | 486 auto padOp = packOp.getSource().getDefiningOp<tensor::PadOp>(); in matchAndRewrite() local
[all...] |
| H A D | ElementwiseOpFusion.cpp | 968 matchAndRewrite(tensor::PadOp padOp,PatternRewriter & rewriter) const matchAndRewrite() argument
1777 matchAndRewrite(tensor::PadOp padOp,PatternRewriter & rewriter) const matchAndRewrite() argument
[all...] |
| /llvm-project/mlir/lib/Dialect/Tensor/Transforms/ |
| H A D | IndependenceTransforms.cpp | 37 FailureOr<Value> tensor::buildIndependentOp(OpBuilder &b, tensor::PadOp padOp, in buildIndependentOp() argument
[all...] |
| H A D | PackAndUnpackPatterns.cpp | 209 Value constantPaddingValue = padOp.getConstantPaddingValue(); in matchAndRewrite() local
|
| H A D | BufferizableOpInterfaceImpl.cpp | 747 padOp.getSource(), options, invocationStack); in getBufferType() local
760 RankedTensorType resultType = padOp.getResultType(); in bufferize() local
[all...] |
| /llvm-project/mlir/lib/Dialect/Tensor/IR/ |
| H A D | ValueBoundsOpInterfaceImpl.cpp | 86 auto padOp = cast<PadOp>(op); populateBoundsForShapedValueDim() local
|
| H A D | TensorInferTypeOpInterfaceImpl.cpp | 171 auto padOp = cast<PadOp>(op); reifyResultShapes() local
|
| H A D | TensorTilingInterfaceImpl.cpp | 30 auto padOp = cast<PadOp>(op); getLoopIteratorTypes() local
577 bubbleUpPadSlice(OpBuilder & b,tensor::PadOp padOp,ArrayRef<OpFoldResult> offsets,ArrayRef<OpFoldResult> sizes,bool generateZeroSliceGuard) bubbleUpPadSlice() argument
[all...] |
| /llvm-project/mlir/lib/Conversion/ArmSMEToSCF/ |
| H A D | ArmSMEToSCF.cpp | 187 auto padOp = tileLoadOp.getPadding(); in matchAndRewrite() local
270 auto padOp = tileLoadOp.getPadding(); in matchAndRewrite() local
|
| /llvm-project/mlir/lib/Dialect/Tensor/TransformOps/ |
| H A D | TensorTransformOps.cpp | 209 if (auto padOp = dyn_cast<tensor::PadOp>(target)) { in applyToOne() local
|
| /llvm-project/mlir/lib/Dialect/Tosa/Transforms/ |
| H A D | TosaValidation.cpp | 41 if (auto padOp = dyn_cast<tosa::PadOp>(op)) { in checkConstantOperandPad() local
|
| /llvm-project/mlir/lib/Conversion/TosaToTensor/ |
| H A D | TosaToTensor.cpp | 305 matchAndRewrite(tosa::PadOp padOp,OpAdaptor adaptor,ConversionPatternRewriter & rewriter) const matchAndRewrite() argument
[all...] |
| /llvm-project/mlir/include/mlir/Dialect/Linalg/Transforms/ |
| H A D | Transforms.h | 484 if (auto padOp = dyn_cast_or_null<tensor::PadOp>(op)) { member
1120 tensor::PadOp padOp; global() member
|
| /llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/Utils/ |
| H A D | LoopEmitter.cpp | 97 auto padOp = t.getDefiningOp<tensor::PadOp>(); in tryFoldTensors() local
210 auto padOp = tensor.getDefiningOp<tensor::PadOp>(); in makeLevelIterator() local
[all...] |
| /llvm-project/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 208 auto padOp = current ? current.getDefiningOp<tensor::PadOp>() : nullptr; in makeComposedPadHighOp() local
|
| /llvm-project/mlir/lib/Dialect/Linalg/TransformOps/ |
| H A D | LinalgTransformOps.cpp | 1956 auto padOp = dyn_cast_or_null<tensor::PadOp>(*targetOps.begin()); apply() local
|