| /llvm-project/mlir/lib/Dialect/Tensor/Transforms/ |
| H A D | PackAndUnpackPatterns.cpp | 445 auto innerDimsPos = unPackOp.getInnerDimsPos(); in matchAndRewrite() local
305 auto innerDimsPos = packOp.getInnerDimsPos(); matchAndRewrite() local
359 auto innerDimsPos = packOp.getInnerDimsPos(); matchAndRewrite() local
403 auto innerDimsPos = unPackOp.getInnerDimsPos(); matchAndRewrite() local
|
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | DataLayoutPropagation.cpp | 77 ArrayRef<int64_t> innerDimsPos = packOrUnPackOp.getInnerDimsPos(); in getPackingInfoFromOperand() local
250 SmallVector<int64_t> innerDimsPos; in getOrCreatePackedViewOfOperand() local
509 ArrayRef<int64_t> innerDimsPos = packOp.getInnerDimsPos(); in matchAndRewrite() local
655 ArrayRef<int64_t> innerDimsPos = packOp.getInnerDimsPos(); bubbleUpPackOpThroughCollapseShape() local
902 ArrayRef<int64_t> innerDimsPos = unPackOp.getInnerDimsPos(); pushDownUnPackOpThroughExpandShape() local
1106 auto innerDimsPos = destPack.getInnerDimsPos(); pushDownUnPackOpThroughGenericOp() local
1173 ArrayRef<int64_t> innerDimsPos = unpackOp.getInnerDimsPos(); matchAndRewrite() local
[all...] |
| H A D | Transforms.cpp | 1084 getPackUnpackRankReducedPerm(ArrayRef<int64_t> shape,ArrayRef<int64_t> innerDimsPos,ArrayRef<int64_t> outerDimsPerm) getPackUnpackRankReducedPerm() argument
1132 auto innerDimsPos = packOp.getInnerDimsPos(); matchAndRewrite() local
1213 ArrayRef<int64_t> innerDimsPos = unpackOp.getInnerDimsPos(); matchAndRewrite() local
|
| H A D | Vectorization.cpp | 1524 auto innerDimsPos = packOp.getInnerDimsPos(); vectorizeAsTensorPackOp() local
|
| /llvm-project/mlir/lib/Dialect/Tensor/Utils/ |
| H A D | Utils.cpp | 83 computePackUnPackPerm(int64_t rank,ArrayRef<int64_t> & innerDimsPos,ArrayRef<int64_t> & outerPerm,PackingMetadata & packingMetadata) computePackUnPackPerm() argument
|
| /llvm-project/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorOps.cpp | 3733 ArrayRef<int64_t> innerDimsPos = packOrUnPack.getInnerDimsPos(); commonVerifierPackAndUnPackOp() local
3808 SmallVector<int64_t> innerDimsPos; global() member
3860 build(OpBuilder & builder,OperationState & state,Value source,Value dest,ArrayRef<int64_t> innerDimsPos,ArrayRef<OpFoldResult> innerTiles,std::optional<Value> paddingValue,ArrayRef<int64_t> outerDimsPerm) build() argument
3897 requirePaddingValue(ArrayRef<int64_t> inputShape,ArrayRef<int64_t> innerDimsPos,ArrayRef<int64_t> outputShape,ArrayRef<int64_t> outerDimsPerm,ArrayRef<OpFoldResult> innerTiles) requirePaddingValue() argument
3971 getPackOpResultTypeShape(ArrayRef<int64_t> sourceShape,ArrayRef<int64_t> innerTileSizes,ArrayRef<int64_t> innerDimsPos,ArrayRef<int64_t> outerDimsPerm) getPackOpResultTypeShape() argument
3995 getResultShape(OpBuilder & builder,Location loc,ArrayRef<OpFoldResult> sourceDims,ArrayRef<OpFoldResult> innerTileSizes,ArrayRef<int64_t> innerDimsPos,ArrayRef<int64_t> outerDimsPerm) getResultShape() argument
4034 inferPackedType(RankedTensorType sourceType,ArrayRef<int64_t> innerTileSizes,ArrayRef<int64_t> innerDimsPos,ArrayRef<int64_t> outerDimsPerm) inferPackedType() argument
4043 createDestinationTensor(OpBuilder & b,Location loc,Value source,ArrayRef<OpFoldResult> innerTileSizes,ArrayRef<int64_t> innerDimsPos,ArrayRef<int64_t> outerDimsPerm) createDestinationTensor() argument
4252 ArrayRef<int64_t> innerDimsPos = packOp.getInnerDimsPos(); isLikePadUnPad() local
4333 build(OpBuilder & builder,OperationState & state,Value source,Value dest,ArrayRef<int64_t> innerDimsPos,ArrayRef<OpFoldResult> innerTiles,ArrayRef<int64_t> outerDimsPerm) build() argument
4352 createDestinationTensor(OpBuilder & b,Location loc,Value source,ArrayRef<OpFoldResult> innerTileSizes,ArrayRef<int64_t> innerDimsPos,ArrayRef<int64_t> outerDimsPerm) createDestinationTensor() argument
[all...] |