Lines Matching refs:computeCostMap
457 DenseMap<Operation *, int64_t> *computeCostMap) { in getComputeCostHelper() argument
464 tripCountOverrideMap, computeCostMap); in getComputeCostHelper()
468 if (computeCostMap != nullptr) { in getComputeCostHelper()
469 auto it = computeCostMap->find(forOp); in getComputeCostHelper()
470 if (it != computeCostMap->end()) { in getComputeCostHelper()
570 DenseMap<Operation *, int64_t> computeCostMap; in getFusionComputeCost() local
596 computeCostMap[insertPointParent] = -storeCount; in getFusionComputeCost()
609 if (computeCostMap.count(forOp) == 0) in getFusionComputeCost()
610 computeCostMap[forOp] = 0; in getFusionComputeCost()
611 computeCostMap[forOp] -= 1; in getFusionComputeCost()
621 srcForOp.getOperation(), srcStats, &sliceTripCountMap, &computeCostMap); in getFusionComputeCost()
624 computeCostMap[insertPointParent] = sliceComputeCost; in getFusionComputeCost()
628 /*tripCountOverrideMap=*/nullptr, &computeCostMap); in getFusionComputeCost()