Searched refs:sliceTripCountMap (Results 1 – 3 of 3) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Affine/Utils/ |
| H A D | LoopFusionUtils.cpp | 454 llvm::SmallDenseMap<Operation *, uint64_t, 8> sliceTripCountMap; in fuseLoops() local
456 return (buildSliceTripCountMap(srcSlice, &sliceTripCountMap) && in fuseLoops()
457 (getSliceIterationCount(sliceTripCountMap) == 1)); in fuseLoops()
581 llvm::SmallDenseMap<Operation *, uint64_t, 8> sliceTripCountMap; in getFusionComputeCost() local
585 if (!buildSliceTripCountMap(slice, &sliceTripCountMap)) in getFusionComputeCost()
588 int64_t sliceIterationCount = getSliceIterationCount(sliceTripCountMap); in getFusionComputeCost()
629 srcForOp, srcStats, &sliceTripCountMap, &computeCostMap); in getFusionComputeCost()
|
| /llvm-project/mlir/include/mlir/Dialect/Affine/Analysis/ |
| H A D | Utils.h | 375 const llvm::SmallDenseMap<Operation *, uint64_t, 8> &sliceTripCountMap);
|
| /llvm-project/mlir/lib/Dialect/Affine/Analysis/ |
| H A D | Utils.cpp | 1640 const llvm::SmallDenseMap<Operation *, uint64_t, 8> &sliceTripCountMap) { in getSliceIterationCount() argument
1642 for (const auto &count : sliceTripCountMap) { in getSliceIterationCount()
1725 llvm::SmallDenseMap<Operation *, uint64_t, 8> sliceTripCountMap; in getComputationSliceState() local
1727 return (buildSliceTripCountMap(*sliceState, &sliceTripCountMap) && in getComputationSliceState()
1728 (getSliceIterationCount(sliceTripCountMap) == 1)); in getComputationSliceState()
|