Searched refs:parallel_strategy (Results 1 – 6 of 6) sorted by relevance
126 … straspb::ParallelStrategy parallel_strategy = parallel_strategys.parallel_strategy(SizeToInt(j)); in Load() local128 size_t dim_num = LongToSize(parallel_strategy.dim_size()); in Load()130 dimension.push_back(parallel_strategy.dim(SizeToInt(k))); in Load()155 straspb::ParallelStrategy *parallel_strategy = parallel_strategys->add_parallel_strategy(); in Save() local156 MS_EXCEPTION_IF_NULL(parallel_strategy); in Save()158 parallel_strategy->add_dim(UlongToUint(LongToUlong(stra_dim))); in Save()
26 ASSERT_TRUE(ConfigManager::GetInstance().parallel_strategy() == ParallelStrategy::ONE_DEVICE); in TEST_F()29 ASSERT_TRUE(ConfigManager::GetInstance().parallel_strategy() == ParallelStrategy::DISTRIBUTION); in TEST_F()
27 repeated ParallelStrategy parallel_strategy = 2; field
79 ParallelStrategy parallel_strategy() const { return parallel_strategy_; } in parallel_strategy() function
62 if (ConfigManager::GetInstance().parallel_strategy() == ParallelStrategy::ONE_DEVICE) { in GraphRunner()
310 if ((ConfigManager::GetInstance().parallel_strategy() == ParallelStrategy::DISTRIBUTION) && in RunGEInitGraph()