Home
last modified time | relevance | path

Searched refs:computation_cost_ (Results 1 – 13 of 13) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dtensor_redistribution.h45 computation_cost_(0.0), in reshape_flag_()
56 double computation_cost() const { return computation_cost_; } in computation_cost()
85 double computation_cost_; variable
Dtensor_redistribution.cc222 computation_cost_ += prod; in ComputeCost()
236 computation_cost_ += COST_FACTOR * prev_prod; in ComputeCost()
255 computation_cost_ += input_size; in ComputePermuteCost()
260 computation_cost_ += (input_size + input_size * dev_num + input_size * dev_num); in ComputePermuteCost()
281 computation_cost_ += input_size; in ComputeConcatCost()
285 computation_cost_ += (input_size + input_size * dev_num + input_size * dev_num); in ComputeConcatCost()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/
Dcostmodel.cc47 return clist_ptrs->at(x)->computation_cost_ < clist_ptrs->at(y)->computation_cost_; in SimplifyForDecreasingCommunicationForward()
70 return clist_ptrs->at(x)->computation_cost_ < clist_ptrs->at(y)->computation_cost_; in SimplifyForDecreasingCommunicationWithPartialPara()
Dedge_costmodel.cc77 MS_LOG(DEBUG) << "The redistribution cost: computation_cost: " << cost->computation_cost_ in InitEdgeCost()
196 recursive(k + 1, computation + c->computation_cost_, memory + c->memory_with_reuse_, in CreateEdgeEliminationCostList()
236 … left_cost->computation_cost_ + middle_cost->computation_cost_ + right_cost->computation_cost_; in CreateOpEliminationSubCostList()
356 … (void)next_op_stras.emplace_back(key_value.first.second, key_value.second[0]->computation_cost_); in GetNextOpStrategyByPrevOpStrategyWithZeroComm()
379 … (void)prev_op_stras.emplace_back(key_value.first.first, key_value.second[0]->computation_cost_); in GetPrevOpStrategyByNextOpStrategyWithZeroComm()
Dgraph_costmodel.cc255 …double computation = cost1->computation_cost_ + cost2->computation_cost_ + cost3->computation_cost… in CreateFinalCostList()
295 …auto new_cost = std::make_shared<Cost>(cost1->computation_cost_, cost1->communication_cost_, decis… in CreateFinalSingleCostList()
339 double minimum = alpha * ret->computation_cost_ + beta * ret->communication_forward_; in SelectCostWithMinInferenceTime()
341 … << "memory_cost: " << ret->memory_with_reuse_ << ", computation_cost_: " << ret->computation_cost_ in SelectCostWithMinInferenceTime()
350 << ", computation_cost_: " << after_mem_filter[i]->computation_cost_ in SelectCostWithMinInferenceTime()
356 …auto tmp = alpha * after_mem_filter[i]->computation_cost_ + beta * after_mem_filter[i]->communicat… in SelectCostWithMinInferenceTime()
394 double minimum = alpha * ret->computation_cost_ + beta * ret->communication_with_partial_para_; in SelectCostWithMinTrainingTime()
396 … << "memory_cost: " << ret->memory_with_reuse_ << ", computation_cost_: " << ret->computation_cost_ in SelectCostWithMinTrainingTime()
404 << ", computation_cost_: " << after_mem_filter[i]->computation_cost_ in SelectCostWithMinTrainingTime()
410 …alpha * after_mem_filter[i]->computation_cost_ + beta * after_mem_filter[i]->communication_with_pa… in SelectCostWithMinTrainingTime()
[all …]
Dcostmodel.h49 …: computation_cost_(computation), communication_cost_(communication), decision_ptr_(std::move(deci… in computation_cost_() function
61 double computation_cost_; member
/third_party/mindspore/tests/ut/cpp/parallel/ops_info/
Dactivation_test.cc88 cost.computation_cost_); in TEST_F()
113 cost.computation_cost_); in TEST_F()
Dtensor_add_info_test.cc192 double memory_cost1 = cost.computation_cost_; in TEST_F()
214 double memory_cost1 = cost.computation_cost_; in TEST_F()
Dtmpidentity_test.cc149 cost.computation_cost_); in TEST_F()
Dmatmul_info_test.cc573 cost.computation_cost_); in TEST_F()
603 cost.computation_cost_); in TEST_F()
/third_party/mindspore/tests/ut/cpp/parallel/auto_parallel/
Dgraph_costmodel_test.cc329 ASSERT_DOUBLE_EQ(ret_list[0]->computation_cost_, 10); in TEST_F()
330 ASSERT_DOUBLE_EQ(ret_list[1]->computation_cost_, 1010); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Doperator_info.cc1426 …if (alpha * s1->cost_list[0]->computation_cost_ + beta * s1->cost_list[0]->communication_with_part… in ApproximateStrategies()
1427 …alpha * s2->cost_list[0]->computation_cost_ + beta * s2->cost_list[0]->communication_with_partial_… in ApproximateStrategies()
1696 if (cost->computation_cost_ > 1.0) { in BreakingTiesForPerferringDataParallel()
1697 cost->computation_cost_ -= 1.0; in BreakingTiesForPerferringDataParallel()
Dmatmul_info.cc626 MS_LOG(DEBUG) << name_ << " : computation_cost: " << result->computation_cost_ in SetCostUnderStrategy()