Home
last modified time | relevance | path

Searched refs:targetShape (Results 1 – 8 of 8) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorUnrollDistribute.cpp34 computeStrides(originalShape, targetShape); in getVectorOffset()
56 : vectorShape(targetShape.begin(), targetShape.end()), in DecomposeShapeIterator()
156 if (!targetShape) in getTargetShape()
162 return targetShape; in getTargetShape()
195 if (!targetShape) in matchAndRewrite()
251 if (!targetShape) in matchAndRewrite()
316 if (!targetShape) in matchAndRewrite()
419 if (!targetShape) in matchAndRewrite()
492 if (!targetShape) in matchAndRewrite()
737 if (!targetShape) in matchAndRewrite()
[all …]
H A DVectorDistribute.cpp344 SmallVector<int64_t> targetShape(writtenVectorType.getShape().begin(), in tryDistributeOp() local
348 if (targetShape[position] % warpOp.getWarpSize() != 0) in tryDistributeOp()
350 targetShape[position] = targetShape[position] / warpOp.getWarpSize(); in tryDistributeOp()
353 VectorType::get(targetShape, writtenVectorType.getElementType()); in tryDistributeOp()
376 auto scale = rewriter.getAffineConstantExpr(targetShape[vectorPos]); in tryDistributeOp()
H A DVectorTransferOpTransforms.cpp219 SmallVector<int64_t> targetShape = llvm::to_vector( in dropUnitDims() local
222 targetShape, inputType, offsets, sizes, strides); in dropUnitDims()
/llvm-project-15.0.7/mlir/lib/Dialect/Utils/
H A DReshapeOpsUtils.cpp32 ArrayRef<int64_t> targetShape) { in getReassociationIndicesForCollapse() argument
33 if (sourceShape.size() <= targetShape.size()) in getReassociationIndicesForCollapse()
37 reassociationMap.reserve(targetShape.size()); in getReassociationIndicesForCollapse()
45 if (targetDim == targetShape.size()) in getReassociationIndicesForCollapse()
48 int64_t currTargetShape = targetShape[targetDim]; in getReassociationIndicesForCollapse()
81 if (reassociationMap.size() != targetShape.size()) in getReassociationIndicesForCollapse()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DVectorization.cpp278 auto targetShape = linalgOp.computeStaticLoopSizes(); in vectorizeLinalgIndex() local
281 llvm::to_vector<16>(llvm::seq<int64_t>(0, targetShape[indexOp.dim()])); in vectorizeLinalgIndex()
287 if (indexOp.dim() == targetShape.size() - 1) in vectorizeLinalgIndex()
292 std::swap(targetShape[indexOp.dim()], targetShape.back()); in vectorizeLinalgIndex()
294 loc, VectorType::get(targetShape, b.getIndexType()), constantOp); in vectorizeLinalgIndex()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Utils/
H A DReshapeOpsUtils.h75 ArrayRef<int64_t> targetShape);
/llvm-project-15.0.7/mlir/include/mlir/Interfaces/
H A DVectorInterfaces.td28 `targetShape`. Return `None` if the op cannot be unrolled to the target
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp359 SmallVector<int64_t> targetShape; in verify() local
365 targetShape.push_back(it.value()); in verify()
367 if (targetShape.empty()) in verify()
371 VectorType::get(targetShape, getSourceVectorType().getElementType()); in verify()