Searched refs:sliceTripCountMap (Results 1 – 3 of 3) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/ |
| H A D | LoopFusionUtils.cpp | 460 llvm::SmallDenseMap<Operation *, uint64_t, 8> sliceTripCountMap; in fuseLoops() local
462 return (buildSliceTripCountMap(srcSlice, &sliceTripCountMap) && in fuseLoops()
463 (getSliceIterationCount(sliceTripCountMap) == 1)); in fuseLoops()
584 llvm::SmallDenseMap<Operation *, uint64_t, 8> sliceTripCountMap; in getFusionComputeCost() local
588 if (!buildSliceTripCountMap(slice, &sliceTripCountMap)) in getFusionComputeCost()
591 int64_t sliceIterationCount = getSliceIterationCount(sliceTripCountMap); in getFusionComputeCost()
636 srcForOp.getOperation(), srcStats, &sliceTripCountMap, &computeCostMap); in getFusionComputeCost()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/Analysis/ |
| H A D | Utils.h | 191 const llvm::SmallDenseMap<Operation *, uint64_t, 8> &sliceTripCountMap);
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/ |
| H A D | Utils.cpp | 1032 const llvm::SmallDenseMap<Operation *, uint64_t, 8> &sliceTripCountMap) { in getSliceIterationCount() argument
1034 for (const auto &count : sliceTripCountMap) { in getSliceIterationCount()
1117 llvm::SmallDenseMap<Operation *, uint64_t, 8> sliceTripCountMap; in getComputationSliceState() local
1119 return (buildSliceTripCountMap(*sliceState, &sliceTripCountMap) && in getComputationSliceState()
1120 (getSliceIterationCount(sliceTripCountMap) == 1)); in getComputationSliceState()
|