Home
last modified time | relevance | path

Searched refs:ExtractStrategy (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.h99 StrategyPtr ExtractStrategy(const ValuePtr &strategy);
Dstep_parallel.cc1316 StrategyPtr ExtractStrategy(const ValuePtr &stra) { in ExtractStrategy() function
1934 strategyPtr = ExtractStrategy(cnode->GetPrimalAttr(STRATEGY)); in ExtractInformation()
1936 strategyPtr = ExtractStrategy(attrs[STRATEGY]); in ExtractInformation()
Dstep_auto_parallel.cc256 strategyPtr = parallel::ExtractStrategy(attrs[STRATEGY]); in SetStrategyToOperator()
/third_party/mindspore/tests/ut/cpp/parallel/
Dstep_parallel_test.cc220 TEST_F(TestStepParallel, ExtractStrategy) { in TEST_F() argument
231 StrategyPtr strategy = ExtractStrategy(attrs["strategy"]); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.cc330 strategy = ExtractStrategy(attrs[STRATEGY]); in CreateOpInfo()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/
Drec_generate_strategy.cc53 StrategyPtr user_defined_stra = parallel::ExtractStrategy(attrs[STRATEGY]); in GenerateStrategy()