Searched refs:return_local_tensor (Results 1 – 2 of 2) sorted by relevance
303 if plan.return_local_tensor is not None:304 for module_path in plan.return_local_tensor:
67 return_local_tensor: Optional[List[str]] = None variable in ShardingPlan