Searched refs:strategy_cost_ (Results 1 – 4 of 4) sorted by relevance
103 std::vector<std::shared_ptr<StrategyWithCost>> GetStrategyCost() { return strategy_cost_; } in GetStrategyCost()124 std::vector<std::shared_ptr<StrategyWithCost>> strategy_cost() const { return strategy_cost_; } in strategy_cost()153 void ClearStrategyCost() { strategy_cost_.clear(); } in ClearStrategyCost()259 std::vector<std::shared_ptr<StrategyWithCost>> strategy_cost_; variable
1399 strategy_cost_.emplace_back(swc); in SetCostUnderStrategyBase()1413 if (strategy_cost_.size() <= target_num) { in ApproximateStrategies()1414 MS_LOG(INFO) << name_ << "'s strategy number is: " << strategy_cost_.size() in ApproximateStrategies()1419 auto &origin_stra_cost = strategy_cost_; in ApproximateStrategies()1437 strategy_cost_ = ret; in ApproximateStrategies()1528 for (auto &swc : strategy_cost_) { in CalculateMemoryCost()1543 for (auto &swc : strategy_cost_) { in CalculateMemoryCostForInference()1551 for (auto &swc : strategy_cost_) { in CorrectMemoryCost()1744 strategy_cost_ = stra_cost; in SetStrategyCost()
414 strategy_cost_.emplace_back(swc); in SetCostForReshape()498 if (strategy_cost_.empty()) { in GenetateStrategyCosts()
465 if (strategy_cost_.empty()) { in GenerateStrategies()637 strategy_cost_.emplace_back(swc); in SetCostUnderStrategy()