Searched refs:expandedShape (Results 1 – 5 of 5) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | ElementwiseOpFusion.cpp | 485 ArrayRef<int64_t> expandedShape,
514 ArrayRef<int64_t> expandedShape, in compute() argument
535 expandedShape.slice(foldedDims.getDimPosition(0), numExpandedDims[pos]); in compute()
569 ArrayRef<int64_t> expandedShape = expansionInfo.getExpandedShapeOfDim(i); in isGenericOpExpandable() local
570 if (expandedShape.size() == 1) in isGenericOpExpandable()
572 for (int64_t shape : expandedShape.drop_front()) { in isGenericOpExpandable()
606 SmallVector<int64_t> expandedShape; in getExpandedType() local
610 expandedShape.append(dimExpansion.begin(), dimExpansion.end()); in getExpandedType()
612 return RankedTensorType::get(expandedShape, originalType.getElementType()); in getExpandedType()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Utils/ |
| H A D | ReshapeOpsUtils.cpp | 228 ArrayRef<int64_t> collapsedShape, ArrayRef<int64_t> expandedShape, in reshapeLikeShapesAreCompatible() argument
235 expandedShape.slice(expandedDimStart, map.value().size()))) { in reshapeLikeShapesAreCompatible()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Math/Transforms/ |
| H A D | PolynomialApproximation.cpp | 111 SmallVector<int64_t> expandedShape(inputShape.begin(), inputShape.end()); in handleMultidimensionalVectors() local
116 expandedShape.insert(expandedShape.end() - 1, expansionDim); in handleMultidimensionalVectors()
117 expandedShape.back() = vectorWidth; in handleMultidimensionalVectors()
122 auto expandedType = VectorType::get(expandedShape, eltType); in handleMultidimensionalVectors()
129 auto iterationDims = ArrayRef<int64_t>(expandedShape).drop_back(); in handleMultidimensionalVectors()
151 Type resultExpandedType = VectorType::get(expandedShape, resultEltType); in handleMultidimensionalVectors()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefOps.cpp | 1614 ArrayRef<int64_t> expandedShape, in verifyCollapsedShape() argument
1634 if (expandedDim >= static_cast<int64_t>(expandedShape.size())) in verifyCollapsedShape()
1639 if (ShapedType::isDynamic(expandedShape[expandedDim])) { in verifyCollapsedShape()
1659 groupSize *= expandedShape[expandedDim]; in verifyCollapsedShape()
1669 for (int64_t d : expandedShape) in verifyCollapsedShape()
1673 } else if (nextDim != static_cast<int64_t>(expandedShape.size())) { in verifyCollapsedShape()
1677 << expandedShape.size() in verifyCollapsedShape()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Utils/ |
| H A D | ReshapeOpsUtils.h | 152 ArrayRef<int64_t> collapsedShape, ArrayRef<int64_t> expandedShape,
|