Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DTiling.cpp55 LoopIndexToRangeIndexMap loopIndexToRangeIndex; in makeTiledLoopRanges() local
63 loopIndexToRangeIndex[idx] = idx - zerosCount; in makeTiledLoopRanges()
71 return std::make_tuple(res, loopIndexToRangeIndex); in makeTiledLoopRanges()
76 const LoopIndexToRangeIndexMap &loopIndexToRangeIndex) { in transformIndexOps() argument
79 auto rangeIndex = loopIndexToRangeIndex.find(en.index()); in transformIndexOps()
80 if (rangeIndex == loopIndexToRangeIndex.end()) in transformIndexOps()
405 LoopIndexToRangeIndexMap loopIndexToRangeIndex; in tileLinalgOpImpl() local
412 if (loopIndexToRangeIndex.count(attr.index())) in tileLinalgOpImpl()
425 auto it = loopIndexToRangeIndex.find(pos); in tileLinalgOpImpl()
426 if (it == loopIndexToRangeIndex.end()) in tileLinalgOpImpl()
[all …]
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Transforms/
H A DTransforms.h533 const LoopIndexToRangeIndexMap &loopIndexToRangeIndex);