Home
last modified time | relevance | path

Searched refs:groupSize (Results 1 – 4 of 4) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Async/Transforms/
H A DAsyncParallelFor.cpp625 Value groupSize = b.create<arith::SubIOp>(blockCount, c1); in doAsyncDispatch() local
626 Value group = b.create<CreateGroupOp>(GroupType::get(ctx), groupSize); in doAsyncDispatch()
663 Value groupSize = b.create<arith::SubIOp>(blockCount, c1); in doSequentialDispatch() local
664 Value group = b.create<CreateGroupOp>(GroupType::get(ctx), groupSize); in doSequentialDispatch()
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/
H A DMemRefOps.cpp1657 int64_t groupSize = 1; in verifyCollapsedShape() local
1659 groupSize *= expandedShape[expandedDim]; in verifyCollapsedShape()
1660 if (groupSize != collapsedShape[collapsedDim]) in verifyCollapsedShape()
1663 << ") must equal reassociation group size (" << groupSize << ")"; in verifyCollapsedShape()
1905 auto groupSize = Wrapper::size(1); in computeCollapsedType() local
1907 groupSize = groupSize * Wrapper::size(srcType.getDimSize(srcDim)); in computeCollapsedType()
1908 resultShape.push_back(groupSize.asSize()); in computeCollapsedType()
/llvm-project-15.0.7/mlir/lib/IR/
H A DAsmPrinter.cpp1207 int groupResultNo = 0, groupSize = 0; in getResultIDAndNumber() local
1212 groupSize = static_cast<int>(owner->getNumResults()) - resultGroups.back(); in getResultIDAndNumber()
1216 groupSize = *it - groupResultNo; in getResultIDAndNumber()
1220 if (groupSize != 1) in getResultIDAndNumber()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/
H A DAffineOps.cpp3529 for (llvm::APInt groupSize : group) { in printMinMaxBound() local
3533 unsigned size = groupSize.getZExtValue(); in printMinMaxBound()