Searched refs:memory_with_reuse_ (Results 1 – 4 of 4) sorted by relevance
256 … double memory = cost1->memory_with_reuse_ + cost2->memory_with_reuse_ + cost3->memory_with_reuse_; in CreateFinalCostList()271 cost->memory_with_reuse_ = memory; in CreateFinalCostList()302 new_cost->memory_with_reuse_ = cost1->memory_with_reuse_; in CreateFinalSingleCostList()323 if (a_cost->memory_with_reuse_ <= memory) { in SelectCostWithMinInferenceTime()325 } else if (a_cost->memory_with_reuse_ < minimum_memory) { in SelectCostWithMinInferenceTime()326 minimum_memory = a_cost->memory_with_reuse_; in SelectCostWithMinInferenceTime()341 … << "memory_cost: " << ret->memory_with_reuse_ << ", computation_cost_: " << ret->computation_cost_ in SelectCostWithMinInferenceTime()349 MS_LOG(INFO) << "Cost " << i << ": memory_cost: " << after_mem_filter[i]->memory_with_reuse_ in SelectCostWithMinInferenceTime()378 if (a_cost->memory_with_reuse_ <= memory) { in SelectCostWithMinTrainingTime()380 } else if (a_cost->memory_with_reuse_ < minimum_memory) { in SelectCostWithMinTrainingTime()[all …]
151 (*cost)->memory_with_reuse_ = mem_cost; in GetRedistributionCost()187 new_cost->memory_with_reuse_ = memory; in CreateEdgeEliminationCostList()196 recursive(k + 1, computation + c->computation_cost_, memory + c->memory_with_reuse_, in CreateEdgeEliminationCostList()245 … left_cost->memory_with_reuse_ + middle_cost->memory_with_reuse_ + right_cost->memory_with_reuse_; in CreateOpEliminationSubCostList()254 cost->memory_with_reuse_ = memory_cost; in CreateOpEliminationSubCostList()310 cost_v[0]->memory_with_reuse_ = 0; in CalculateMemoryCost()327 cost_v[0]->memory_with_reuse_ = 0; in CalculateMemoryCostForInference()
50 memory_with_reuse_ = 0.0; in computation_cost_()58 double memory_with_reuse_; member
1530 swc->cost_list[0]->memory_with_reuse_ = mem_cost; in CalculateMemoryCost()1545 swc->cost_list[0]->memory_with_reuse_ = mem_cost; in CalculateMemoryCostForInference()1554 swc->cost_list[0]->memory_with_reuse_ -= parameter_mem_cost; in CorrectMemoryCost()1555 if (swc->cost_list[0]->memory_with_reuse_ < 0) { in CorrectMemoryCost()1556 … MS_LOG(ERROR) << "The memory cost after correction is: " << swc->cost_list[0]->memory_with_reuse_ in CorrectMemoryCost()