Home
last modified time | relevance | path

Searched defs:outerDimsPerm (Results 1 – 5 of 5) sorted by relevance

/llvm-project/mlir/lib/Dialect/Tensor/Transforms/
H A DPackAndUnpackPatterns.cpp98 auto outerDimsPerm = packOp.getOuterDimsPerm(); isPackOnInnerMostDim() local
158 auto outerDimsPerm = unpackOp.getOuterDimsPerm(); isUnpackOnInnerMostDim() local
307 auto outerDimsPerm = packOp.getOuterDimsPerm(); matchAndRewrite() local
358 auto outerDimsPerm = packOp.getOuterDimsPerm(); matchAndRewrite() local
402 auto outerDimsPerm = unPackOp.getOuterDimsPerm(); matchAndRewrite() local
444 auto outerDimsPerm = unPackOp.getOuterDimsPerm(); matchAndRewrite() local
[all...]
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DDataLayoutPropagation.cpp177 SmallVector<int64_t> outerDimsPerm; in computeOuterDims() local
262 SmallVector<int64_t> outerDimsPerm; in getOrCreatePackedViewOfOperand() local
523 ArrayRef<int64_t> outerDimsPerm = packOp.getOuterDimsPerm(); matchAndRewrite() local
656 ArrayRef<int64_t> outerDimsPerm = packOp.getOuterDimsPerm(); bubbleUpPackOpThroughCollapseShape() local
763 ArrayRef<int64_t> outerDimsPerm = packOp.getOuterDimsPerm(); bubbleUpPackOpThroughExpandShape() local
903 ArrayRef<int64_t> outerDimsPerm = unPackOp.getOuterDimsPerm(); pushDownUnPackOpThroughExpandShape() local
1107 auto outerDimsPerm = destPack.getOuterDimsPerm(); pushDownUnPackOpThroughGenericOp() local
1185 ArrayRef<int64_t> outerDimsPerm = unpackOp.getOuterDimsPerm(); matchAndRewrite() local
[all...]
H A DVectorization.cpp1525 auto outerDimsPerm = packOp.getOuterDimsPerm(); vectorizeAsTensorPackOp() local
1580 ArrayRef<int64_t> outerDimsPerm = unpackOp.getOuterDimsPerm(); vectorizeAsTensorUnpackOp() local
H A DTransforms.cpp1085 getPackUnpackRankReducedPerm(ArrayRef<int64_t> shape,ArrayRef<int64_t> innerDimsPos,ArrayRef<int64_t> outerDimsPerm) getPackUnpackRankReducedPerm() argument
/llvm-project/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp3810 SmallVector<int64_t> outerDimsPerm; global() member
3863 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
3899 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
3996 getResultShape(OpBuilder & builder,Location loc,ArrayRef<OpFoldResult> sourceDims,ArrayRef<OpFoldResult> innerTileSizes,ArrayRef<int64_t> innerDimsPos,ArrayRef<int64_t> outerDimsPerm) getResultShape() argument
4035 inferPackedType(RankedTensorType sourceType,ArrayRef<int64_t> innerTileSizes,ArrayRef<int64_t> innerDimsPos,ArrayRef<int64_t> outerDimsPerm) inferPackedType() argument
4044 createDestinationTensor(OpBuilder & b,Location loc,Value source,ArrayRef<OpFoldResult> innerTileSizes,ArrayRef<int64_t> innerDimsPos,ArrayRef<int64_t> outerDimsPerm) createDestinationTensor() argument
4335 build(OpBuilder & builder,OperationState & state,Value source,Value dest,ArrayRef<int64_t> innerDimsPos,ArrayRef<OpFoldResult> innerTiles,ArrayRef<int64_t> outerDimsPerm) build() argument
4353 createDestinationTensor(OpBuilder & b,Location loc,Value source,ArrayRef<OpFoldResult> innerTileSizes,ArrayRef<int64_t> innerDimsPos,ArrayRef<int64_t> outerDimsPerm) createDestinationTensor() argument
[all...]