| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | DataLayoutPropagation.cpp | 374 bubbleUpPackOpThroughGenericOp(RewriterBase &rewriter, tensor::PackOp packOp, in bubbleUpPackOpThroughGenericOp() argument
484 LogicalResult matchAndRewrite(tensor::PackOp packOp, in matchAndRewrite() argument
652 bubbleUpPackOpThroughCollapseShape(tensor::CollapseShapeOp collapseOp,tensor::PackOp packOp,PatternRewriter & rewriter) bubbleUpPackOpThroughCollapseShape() argument
759 bubbleUpPackOpThroughExpandShape(tensor::ExpandShapeOp expandOp,tensor::PackOp packOp,PatternRewriter & rewriter) bubbleUpPackOpThroughExpandShape() argument
842 matchAndRewrite(tensor::PackOp packOp,PatternRewriter & rewriter) const matchAndRewrite() argument
[all...] |
| H A D | BlockPackMatmul.cpp | 91 tensor::PackOp packOp, AffineMap operandMap, in transposePackedMatmul() argument
|
| H A D | Transforms.cpp | 220 tensor::PackOp packOp, in lowerPack() argument
678 packTranspose(RewriterBase & rewriter,tensor::PackOp packOp,linalg::LinalgOp linalgOp,tensor::UnPackOp maybeUnPackOp,ArrayRef<int64_t> outerPerm,ArrayRef<int64_t> innerPerm) packTranspose() argument
1030 getPackOpSourceOrPaddedSource(OpBuilder & builder,tensor::PackOp packOp) getPackOpSourceOrPaddedSource() argument
1123 matchAndRewrite(tensor::PackOp packOp,PatternRewriter & rewriter) const matchAndRewrite() argument
[all...] |
| H A D | Vectorization.cpp | 1408 getTiledPackShape(tensor::PackOp packOp,ArrayRef<int64_t> destShape) getTiledPackShape() argument
1492 vectorizeAsTensorPackOp(RewriterBase & rewriter,tensor::PackOp packOp,ArrayRef<int64_t> inputVectorSizes,SmallVectorImpl<Value> & newResults) vectorizeAsTensorPackOp() argument
1877 vectorizePackOpPrecondition(tensor::PackOp packOp,ArrayRef<int64_t> inputVectorSizes) vectorizePackOpPrecondition() argument
1973 __anon729377362102(auto packOp) vectorizeOpPrecondition() argument
2070 __anon729377362602(auto packOp) vectorize() argument
[all...] |
| /llvm-project/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorTilingInterfaceImpl.cpp | 109 auto packOp = cast<PackOp>(op); getLoopIteratorTypes() local
123 auto packOp = cast<PackOp>(op); getTiledImplementation() local
208 auto packOp = cast<PackOp>(op); getResultTilePosition() local
228 auto packOp = cast<PackOp>(op); generateResultTileValue() local
[all...] |
| H A D | TensorOps.cpp | 4119 hasSameInnerOuterAttribute(PackOp packOp,UnPackOp unPackOp) hasSameInnerOuterAttribute() argument
4133 haveSameTiles(PackOp packOp,UnPackOp unPackOp) haveSameTiles() argument
4160 inferStaticShape(PackOp packOp,SmallVectorImpl<int64_t> & srcShape,SmallVectorImpl<int64_t> & destShape) inferStaticShape() argument
4195 canonicalize(PackOp packOp,PatternRewriter & rewriter) canonicalize() argument
4244 isLikePadUnPad(PackOrUnpackOp packOp,RankedTensorType packedTensorType) isLikePadUnPad() argument
4431 if (PackOp packOp = unPackOp.getSource().getDefiningOp<tensor::PackOp>()) { canonicalize() local
[all...] |
| /llvm-project/mlir/lib/Dialect/Tensor/Utils/ |
| H A D | Utils.cpp | 104 getPackInverseDestPerm(PackOp packOp) getPackInverseDestPerm() argument
|
| /llvm-project/mlir/lib/Dialect/Tensor/Transforms/ |
| H A D | PackAndUnpackPatterns.cpp | 295 auto packOp = linalgOp->getOperand(0).getDefiningOp<PackOp>(); matchAndRewrite() local
[all...] |
| /llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgOps.cpp | 812 foldFillPackIntoFillOp(RewriterBase & rewriter,tensor::PackOp packOp) foldFillPackIntoFillOp() argument
|
| /llvm-project/mlir/lib/Dialect/Linalg/TransformOps/ |
| H A D | LinalgTransformOps.cpp | 1637 auto packOp = dyn_cast<tensor::PackOp>(*packOrUnpackOps.begin()); apply() local
[all...] |