Searched refs:dstShape (Results 1 – 3 of 3) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | VectorUnrollDistribute.cpp | 380 SmallVector<int64_t> dstShape = in matchAndRewrite() local
382 auto targetType = VectorType::get(dstShape, dstVecType.getElementType()); in matchAndRewrite()
439 SmallVector<int64_t> dstShape; in matchAndRewrite() local
444 dstShape.push_back((*targetShape)[i]); in matchAndRewrite()
456 loc, reductionOp.getAcc(), destOffset, dstShape, accStrides); in matchAndRewrite()
459 dstShape, reductionOp.getSourceVectorType().getElementType()); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/ |
| H A D | TosaToLinalg.cpp | 914 ArrayRef<int64_t> dstShape, in createReassociationMapsForCollapse() argument
926 if (dstShape.empty()) { in createReassociationMapsForCollapse()
931 reassociationMap.resize(dstShape.size()); in createReassociationMapsForCollapse()
933 while (currSrcDim < srcShape.size() && currDstDim < dstShape.size()) { in createReassociationMapsForCollapse()
934 int64_t dstSize = dstShape[currDstDim]; in createReassociationMapsForCollapse()
946 if (currDstDim == dstShape.size() - 1 || dstShape[currDstDim + 1] != 1) { in createReassociationMapsForCollapse()
958 return currSrcDim == srcShape.size() && currDstDim == dstShape.size(); in createReassociationMapsForCollapse()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/MemRefToLLVM/ |
| H A D | MemRefToLLVM.cpp | 1602 SmallVector<Value> dstShape = getDynamicOutputShape( in matchAndRewrite() local
1605 for (auto &en : llvm::enumerate(dstShape)) in matchAndRewrite()
1617 llvm::reverse(llvm::seq<int64_t>(0, dstShape.size()))) { in matchAndRewrite()
1619 stride = rewriter.create<LLVM::MulOp>(loc, dstShape[dimIndex], stride); in matchAndRewrite()
|