Searched refs:_test_simple_model_with_meta_device (Results 1 – 1 of 1) sorted by relevance
153 def _test_simple_model_with_meta_device(self, meta_module_fn, init_fn=None): member in TestFSDPWithMetaDevice202 self._test_simple_model_with_meta_device(211 self._test_simple_model_with_meta_device(meta_module_fn)222 self._test_simple_model_with_meta_device(meta_module_fn)233 self._test_simple_model_with_meta_device(