Home
last modified time | relevance | path

Searched defs:reassociation (Results 1 – 14 of 14) sorted by relevance

/llvm-project/mlir/lib/Dialect/Tensor/IR/
H A DTensorInferTypeOpInterfaceImpl.cpp23 getExpandedDimToCollapsedDimMap(ArrayRef<AffineMap> reassociation) getExpandedDimToCollapsedDimMap() argument
71 getCollapsedOutputShapeFromInputShape(OpBuilder & builder,Location loc,Value src,ArrayRef<int64_t> dstStaticShape,ArrayRef<AffineMap> reassociation) getCollapsedOutputShapeFromInputShape() argument
83 getExpandedOutputDimFromInputShape(OpBuilder & builder,Location loc,int64_t dimIndex,Value src,ArrayRef<int64_t> dstStaticShape,ArrayRef<AffineMap> reassociation,llvm::DenseMap<int64_t,int64_t> & expandedDimToCollapsedDim) getExpandedOutputDimFromInputShape() argument
123 getExpandedOutputShapeFromInputShape(OpBuilder & builder,Location loc,Value src,ArrayRef<int64_t> dstStaticShape,ArrayRef<AffineMap> reassociation) getExpandedOutputShapeFromInputShape() argument
H A DTensorOps.cpp1654 inferOutputShape(OpBuilder & b,Location loc,RankedTensorType expandedType,ArrayRef<ReassociationIndices> reassociation,ArrayRef<OpFoldResult> inputShape) inferOutputShape() argument
1666 build(OpBuilder & builder,OperationState & result,Type resultType,Value src,ArrayRef<ReassociationIndices> reassociation,ArrayRef<OpFoldResult> outputShape) build() argument
1677 build(OpBuilder & builder,OperationState & result,Type resultType,Value src,ArrayRef<ReassociationIndices> reassociation) build() argument
1708 inferCollapsedType(RankedTensorType type,SmallVector<ReassociationIndices> reassociation) inferCollapsedType() argument
1718 inferCollapsedType(RankedTensorType type,ArrayRef<AffineMap> reassociation) inferCollapsedType() argument
1744 build(OpBuilder & b,OperationState & result,Value src,ArrayRef<ReassociationIndices> reassociation,ArrayRef<NamedAttribute> attrs) build() argument
[all...]
/llvm-project/mlir/lib/Dialect/Utils/
H A DReshapeOpsUtils.cpp161 OpBuilder &b, ArrayRef<ReassociationIndices> reassociation) { in getReassociationIndicesAttribute() argument
184 mlir::getSymbolLessAffineMaps(ArrayRef<ReassociationExprs> reassociation) { in getSymbolLessAffineMaps() argument
197 bool mlir::isReassociationValid(ArrayRef<AffineMap> reassociation, in isReassociationValid() argument
431 SmallVector<int64_t, 2> reassociation; getSimplifyCollapseShapeWithRankReducingSliceInfo() local
[all...]
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DDropUnitDims.cpp258 expandValue(RewriterBase & rewriter,Location loc,Value result,Value origDest,ArrayRef<ReassociationIndices> reassociation,ControlDropUnitDims::RankReductionStrategy rankReductionStrategy) expandValue() argument
286 collapseValue(RewriterBase & rewriter,Location loc,Value operand,ArrayRef<int64_t> targetShape,ArrayRef<ReassociationIndices> reassociation,ControlDropUnitDims::RankReductionStrategy rankReductionStrategy) collapseValue() argument
337 SmallVector<ReassociationIndices> reassociation; global() member
700 auto reassociation = getReassociationMapForFoldingUnitDims(targetShape); matchAndRewrite() local
734 auto reassociation = getReassociationMapForFoldingUnitDims(targetShape); matchAndRewrite() local
845 SmallVector<ReassociationIndices> reassociation(rank - 1, {0, 1}); getReassociationForReshapeAtDim() local
[all...]
H A DElementwiseOpFusion.cpp568 SmallVector<ReassociationIndices> reassociation; global() member in __anona62ab6fa0911::ExpansionInfo
691 SmallVector<ReassociationIndices> reassociation; getReassociationForExpansion() local
834 SmallVector<ReassociationIndices> reassociation = fuseWithReshapeByExpansion() local
859 SmallVector<ReassociationIndices> reassociation = fuseWithReshapeByExpansion() local
901 SmallVector<ReassociationIndices> reassociation = fuseWithReshapeByExpansion() local
1216 getCollapsableIterationSpaceDims(GenericOp genericOp,OpOperand * fusableOperand,ArrayRef<ReassociationIndices> reassociation) getCollapsableIterationSpaceDims() argument
1702 SmallVector<ReassociationIndices> reassociation = collapseOpIterationDims() local
[all...]
H A DSplitReduction.cpp81 SmallVector<ReassociationIndices> reassociation; in splitReduction() local
/llvm-project/mlir/lib/Dialect/Tensor/Transforms/
H A DPackAndUnpackPatterns.cpp128 auto reassociation = matchAndRewrite() local
190 auto reassociation = matchAndRewrite() local
[all...]
H A DExtractSliceFromReshapeUtils.cpp51 OpBuilder &b, Location loc, ArrayRef<ReassociationIndices> reassociation, in invertCollapseShapeIndexing()
/llvm-project/mlir/lib/Dialect/MemRef/IR/
H A DMemRefOps.cpp2115 verifyCollapsedShape(Operation * op,ArrayRef<int64_t> collapsedShape,ArrayRef<int64_t> expandedShape,ArrayRef<ReassociationIndices> reassociation,bool allowMultipleDynamicDimsPerGroup) verifyCollapsedShape() argument
2207 computeExpandedLayoutMap(MemRefType srcType,ArrayRef<int64_t> resultShape,ArrayRef<ReassociationIndices> reassociation) computeExpandedLayoutMap() argument
2247 computeExpandedType(MemRefType srcType,ArrayRef<int64_t> resultShape,ArrayRef<ReassociationIndices> reassociation) computeExpandedType() argument
2268 inferOutputShape(OpBuilder & b,Location loc,MemRefType expandedType,ArrayRef<ReassociationIndices> reassociation,ArrayRef<OpFoldResult> inputShape) inferOutputShape() argument
2280 build(OpBuilder & builder,OperationState & result,Type resultType,Value src,ArrayRef<ReassociationIndices> reassociation,ArrayRef<OpFoldResult> outputShape) build() argument
2291 build(OpBuilder & builder,OperationState & result,Type resultType,Value src,ArrayRef<ReassociationIndices> reassociation) build() argument
2305 build(OpBuilder & builder,OperationState & result,ArrayRef<int64_t> resultShape,Value src,ArrayRef<ReassociationIndices> reassociation) build() argument
2318 build(OpBuilder & builder,OperationState & result,ArrayRef<int64_t> resultShape,Value src,ArrayRef<ReassociationIndices> reassociation,ArrayRef<OpFoldResult> outputShape) build() argument
2402 computeCollapsedLayoutMap(MemRefType srcType,ArrayRef<ReassociationIndices> reassociation,bool strict=false) computeCollapsedLayoutMap() argument
2460 isGuaranteedCollapsible(MemRefType srcType,ArrayRef<ReassociationIndices> reassociation) isGuaranteedCollapsible() argument
2470 computeCollapsedType(MemRefType srcType,ArrayRef<ReassociationIndices> reassociation) computeCollapsedType() argument
2501 build(OpBuilder & b,OperationState & result,Value src,ArrayRef<ReassociationIndices> reassociation,ArrayRef<NamedAttribute> attrs) build() argument
[all...]
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/Utils/
H A DCodegenUtils.cpp222 ArrayRef<ReassociationIndices> reassociation) { in genReshapeDstShape() argument
277 ArrayRef<ReassociationIndices> reassociation, // NOLINT in reshapeCvs() argument
/llvm-project/mlir/lib/Dialect/Arith/Utils/
H A DUtils.cpp26 ArrayRef<ReassociationIndices> reassociation, in inferExpandShapeOutputShape() argument
/llvm-project/mlir/lib/Dialect/Vector/Transforms/
H A DVectorTransferOpTransforms.cpp498 SmallVector<ReassociationIndices> reassociation; collapseInnerDims() local
/llvm-project/mlir/lib/Dialect/Linalg/Utils/
H A DUtils.cpp887 SmallVector<ReassociationIndices> reassociation; getReassociationMapForFoldingUnitDims() local
/llvm-project/mlir/lib/Dialect/Linalg/TransformOps/
H A DLinalgTransformOps.cpp3518 ReassociationIndices reassociation(target.getNumLoops()); applyToOne() local