Searched refs:sourceShape (Results 1 – 5 of 5) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Utils/ |
| H A D | ReshapeOpsUtils.cpp | 31 mlir::getReassociationIndicesForCollapse(ArrayRef<int64_t> sourceShape, in getReassociationIndicesForCollapse() argument
33 if (sourceShape.size() <= targetShape.size()) in getReassociationIndicesForCollapse()
41 while (sourceDim < sourceShape.size()) { in getReassociationIndicesForCollapse()
49 while (sourceShape[sourceDim] != ShapedType::kDynamicSize && in getReassociationIndicesForCollapse()
51 sourceDim < sourceShape.size()) { in getReassociationIndicesForCollapse()
52 prodOfCollapsedDims *= sourceShape[sourceDim]; in getReassociationIndicesForCollapse()
59 if (sourceShape[sourceDim] == ShapedType::kDynamicSize && in getReassociationIndicesForCollapse()
67 sourceShape[sourceDim] != ShapedType::kDynamicSize) in getReassociationIndicesForCollapse()
85 for (; sourceDim < sourceShape.size(); sourceDim++) { in getReassociationIndicesForCollapse()
86 if (sourceShape[sourceDim] != ShapedType::kDynamicSize && in getReassociationIndicesForCollapse()
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgOps.cpp | 1778 ArrayRef<int64_t> sourceShape = sourceType.getShape(); in populateMap() local
1784 sourceShape = castSourceType.getShape(); in populateMap()
1790 for (unsigned i = 0; i < sourceShape.size(); i++) { in populateMap()
1794 affineExprToSize.try_emplace(affineDimExpr, sourceShape[i]); in populateMap()
1819 ArrayRef<int64_t> sourceShape = sourceType.getShape(); in createNewOperandWithStaticSizes() local
1825 for (unsigned i = 0; i < sourceShape.size(); i++) { in createNewOperandWithStaticSizes()
1826 int64_t dimShape = sourceShape[i]; in createNewOperandWithStaticSizes()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Utils/ |
| H A D | ReshapeOpsUtils.h | 74 getReassociationIndicesForCollapse(ArrayRef<int64_t> sourceShape,
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/ |
| H A D | VectorOps.cpp | 2207 auto sourceShape = sourceVectorType.getShape(); in verify() local
2210 destShape.size() - sourceShape.size(), 0); in verify()
2211 sourceShapeAsDestShape.append(sourceShape.begin(), sourceShape.end()); in verify()
4288 auto sourceShape = sourceVectorType.getShape(); in verifyVectorShapeCast() local
4293 sourceShape.begin(), sourceShape.end(), 1LL, std::multiplies<int64_t>{}); in verifyVectorShapeCast()
4303 if (!isValidShapeCast(sourceShape, resultShape)) in verifyVectorShapeCast()
4306 if (!isValidShapeCast(resultShape, sourceShape)) in verifyVectorShapeCast()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefOps.cpp | 2502 ArrayRef<int64_t> sourceShape = subViewOp.getSourceType().getShape(); in isTrivialSubViewOp() local
2505 if (!intValue || *intValue != sourceShape[size.index()]) in isTrivialSubViewOp()
|