Searched refs:sharding_propagation (Results 1 – 10 of 10) sorted by relevance
55 sharding_propagation=True)64 sharding_propagation=True)72 sharding_propagation=True)81 sharding_propagation=True)
57 sharding_propagation=True)67 sharding_propagation=True)
73 context.set_auto_parallel_context(sharding_propagation=True)76 context.set_auto_parallel_context(sharding_propagation=False)
55 sharding_propagation=True)
292 sharding_propagation=True)
293 sharding_propagation=True)
488 def set_sharding_propagation(self, sharding_propagation): argument499 if not isinstance(sharding_propagation, bool):501 self._context_handle.set_sharding_propagation(sharding_propagation)687 sharding_propagation=bool, enable_alltoall=bool)
131 bool sharding_propagation() const { return sharding_propagation_; } in sharding_propagation() function
357 if (ParallelContext::GetInstance()->sharding_propagation() && in CreateTheOperatorInfo()630 if (ParallelContext::GetInstance()->sharding_propagation() && (prev_prim->name() == CAST) && in CreateEdgeBetweenTwoOps()969 if (ParallelContext::GetInstance()->sharding_propagation()) { in ParallelStrategySearch()
201 ….def("get_sharding_propagation", &ParallelContext::sharding_propagation, "Get sharding strategy pr… in PYBIND11_MODULE()