Searched refs:_test_multi_forward_module (Results 1 – 1 of 1) sorted by relevance
462 self._test_multi_forward_module,465 def _test_multi_forward_module(self, reshard_after_forward: Union[bool, int]): member in TestFullyShard1DTrainingCore