Searched refs:set_dataset_strategy (Results 1 – 4 of 4) sorted by relevance
266 def set_dataset_strategy(self, dataset_strategy): member in _AutoParallelContext290 self._context_handle.set_dataset_strategy(dataset_strategy)645 "dataset_strategy": auto_parallel_context().set_dataset_strategy,
68 void set_dataset_strategy(const std::vector<std::vector<int64_t>> &dataset_strategy);
94 void ParallelContext::set_dataset_strategy(const std::vector<std::vector<int64_t>> &dataset_strateg… in set_dataset_strategy() function in mindspore::parallel::ParallelContext
183 ….def("set_dataset_strategy", &ParallelContext::set_dataset_strategy, "Set dataset sharding strateg… in PYBIND11_MODULE()