Searched refs:set_enable_all2all (Results 1 – 3 of 3) sorted by relevance
132 void set_enable_all2all(const bool);
280 void ParallelContext::set_enable_all2all(const bool enable) { enable_all2all_ = enable; } in set_enable_all2all() function in mindspore::parallel::ParallelContext
202 ….def("set_enable_alltoall", &ParallelContext::set_enable_all2all, "Set the enabling AllToAll value… in PYBIND11_MODULE()