Searched defs:expandedShape (Results 1 – 4 of 4) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | ElementwiseOpFusion.cpp | 581 compute(LinalgOp linalgOp,OpOperand * fusableOpOperand,ArrayRef<AffineMap> reassociationMaps,ArrayRef<int64_t> expandedShape,ArrayRef<int64_t> collapsedShape,PatternRewriter & rewriter) compute() argument
636 ArrayRef<int64_t> expandedShape = expansionInfo.getExpandedShapeOfDim(i); isLinalgOpExpandable() local
673 SmallVector<int64_t> expandedShape; getExpandedType() local
754 ArrayRef<int64_t> expandedShape = expansionInfo.getExpandedShapeOfDim(i); validateDynamicDimExpansion() local
[all...] |
| /llvm-project/mlir/lib/Dialect/Utils/ |
| H A D | ReshapeOpsUtils.cpp | 229 ArrayRef<int64_t> collapsedShape, ArrayRef<int64_t> expandedShape, in reshapeLikeShapesAreCompatible() argument
|
| /llvm-project/mlir/lib/Dialect/Math/Transforms/ |
| H A D | PolynomialApproximation.cpp | 125 SmallVector<int64_t> expandedShape(inputShape.begin(), inputShape.end()); handleMultidimensionalVectors() local
|
| /llvm-project/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefOps.cpp | 2114 verifyCollapsedShape(Operation * op,ArrayRef<int64_t> collapsedShape,ArrayRef<int64_t> expandedShape,ArrayRef<ReassociationIndices> reassociation,bool allowMultipleDynamicDimsPerGroup) verifyCollapsedShape() argument
|