Searched defs:auto_wrap (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/fsdp/ |
| H A D | test_fsdp_meta.py | 238 self, auto_wrap, meta_module_fn, init_fn=None argument
291 def test_nested_model_with_meta_device_reset_params(self, auto_wrap): argument
303 def test_nested_model_with_meta_device_default_init(self, auto_wrap): argument
318 def test_nested_model_with_torchdistX_default_init(self, auto_wrap): argument
332 def test_nested_model_with_torchdistX_init_fn(self, auto_wrap): argument
|
| /aosp_15_r20/tools/asuite/atest/ |
| H A D | atest_utils.py | 701 def colorful_print(text, color=None, bp_color=None, auto_wrap=True): argument
|