Searched defs:auto_wrap (Results 1 – 1 of 1) sorted by relevance
238 self, auto_wrap, meta_module_fn, init_fn=None argument291 def test_nested_model_with_meta_device_reset_params(self, auto_wrap): argument303 def test_nested_model_with_meta_device_default_init(self, auto_wrap): argument318 def test_nested_model_with_torchdistX_default_init(self, auto_wrap): argument332 def test_nested_model_with_torchdistX_init_fn(self, auto_wrap): argument