Home
last modified time | relevance | path

Searched refs:forward_hook_wrong_input1 (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_hooks.py326 def forward_hook_wrong_input1(self, input: Tuple[str, str], output: str): function
330 m.register_forward_hook(forward_hook_wrong_input1)