Lines Matching refs:computeCostMap
536 DenseMap<Operation *, int64_t> *computeCostMap) { in getComputeCostHelper() argument
543 tripCountOverrideMap, computeCostMap); in getComputeCostHelper()
547 if (computeCostMap != nullptr) { in getComputeCostHelper()
548 auto it = computeCostMap->find(forOp); in getComputeCostHelper()
549 if (it != computeCostMap->end()) { in getComputeCostHelper()
585 DenseMap<Operation *, int64_t> computeCostMap; in getFusionComputeCost() local
611 computeCostMap[insertPointParent] = -storeCount; in getFusionComputeCost()
624 if (computeCostMap.count(forOp) == 0) in getFusionComputeCost()
625 computeCostMap[forOp] = 0; in getFusionComputeCost()
626 computeCostMap[forOp] -= 1; in getFusionComputeCost()
636 srcForOp.getOperation(), srcStats, &sliceTripCountMap, &computeCostMap); in getFusionComputeCost()
639 computeCostMap[insertPointParent] = sliceComputeCost; in getFusionComputeCost()
643 /*tripCountOverrideMap=*/nullptr, &computeCostMap); in getFusionComputeCost()