Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/parallel/
D_auto_parallel_context.py488 def set_sharding_propagation(self, sharding_propagation): member in _AutoParallelContext
501 self._context_handle.set_sharding_propagation(sharding_propagation)
652 "sharding_propagation": auto_parallel_context().set_sharding_propagation,
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dcontext.h130 void set_sharding_propagation(const bool);
Dcontext.cc278 void ParallelContext::set_sharding_propagation(const bool stra_pto) { sharding_propagation_ = stra_… in set_sharding_propagation() function in mindspore::parallel::ParallelContext
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dinit.cc199 .def("set_sharding_propagation", &ParallelContext::set_sharding_propagation, in PYBIND11_MODULE()