Searched refs:my_pre_load_hook_with_module (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/nn/ | ||
| H A D | test_module_hooks.py | 651 def my_pre_load_hook_with_module( member in TestStateDictHooks.test_load_state_dict_module_pre_hook.MyModule |