Searched refs:_get_optimize_ddp_mode (Results 1 – 3 of 3) sorted by relevance
294 def _get_optimize_ddp_mode(): function
1245 if config._get_optimize_ddp_mode() == "ddp_optimizer":
900 optimize_ddp = torch._dynamo.config._get_optimize_ddp_mode()