Home
last modified time | relevance | path

Searched refs:s_strategy (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Doperator_info.h140 void SetSelectedStrategyAndCost(const StrategyPtr &s_strategy, const CostPtr &cost) { in SetSelectedStrategyAndCost() argument
141 selected_strategy_ = s_strategy; in SetSelectedStrategyAndCost()
144 void SetSelectedStrategy(const StrategyPtr &s_strategy, size_t);
155 Status InitSelectedStrategy(const StrategyPtr &s_strategy) { return Init(s_strategy); } in InitSelectedStrategy() argument
Doperator_info.cc1712 void OperatorInfo::SetSelectedStrategy(const StrategyPtr &s_strategy, size_t curr_depth) { in SetSelectedStrategy() argument
1713 MS_EXCEPTION_IF_NULL(s_strategy); in SetSelectedStrategy()
1719 PrintStrategy(s_strategy); in SetSelectedStrategy()
1720 selected_strategy_ = s_strategy; in SetSelectedStrategy()
1733 void OperatorInfo::CheckSelectedStrategy(const StrategyPtr &s_strategy) { in CheckSelectedStrategy() argument
1734 MS_EXCEPTION_IF_NULL(s_strategy); in CheckSelectedStrategy()
1735 if (!s_strategy->IsEqual(selected_strategy_)) { in CheckSelectedStrategy()
1739 PrintStrategy(s_strategy); in CheckSelectedStrategy()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_auto_parallel.cc986 StrategyPtr s_strategy = op->selected_strategy(); in ParallelStrategySearch() local
988 PrintStrategy(s_strategy); in ParallelStrategySearch()
1104 StrategyPtr s_strategy = op->selected_strategy(); in ParallelStrategyRecSearch() local
1106 PrintStrategy(s_strategy); in ParallelStrategyRecSearch()
/third_party/mindspore/tests/ut/cpp/parallel/auto_parallel/
Ddp_algo_test.cc1328 StrategyPtr s_strategy = op->selected_strategy(); in TEST_F() local
1329 Dimensions strategy_0 = s_strategy->GetInputDim()[0]; in TEST_F()
1330 Dimensions strategy_1 = s_strategy->GetInputDim()[1]; in TEST_F()