Home
last modified time | relevance | path

Searched refs:sliceDimCounts (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Utils/
H A DVectorUtils.cpp58 SmallVector<int64_t, 4> sliceDimCounts(rank); in computeStrides() local
60 sliceDimCounts[r] = ceilDiv(shape[r], sizes[r]); in computeStrides()
65 sliceStrides[r] = sliceStrides[r + 1] * sliceDimCounts[r + 1]; in computeStrides()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorUnrollDistribute.cpp63 SmallVector<int64_t> sliceDimCounts(originalShape.size()); in DecomposeShapeIterator() local
65 sliceDimCounts[r] = ceilDiv(originalShape[r], targetShape[r]); in DecomposeShapeIterator()
66 maxIndexVal *= sliceDimCounts[r]; in DecomposeShapeIterator()
74 accum *= sliceDimCounts[idx]; in DecomposeShapeIterator()