Searched defs:fw_hook (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/test/nn/ |
| H A D | test_module_hooks.py | 517 def fw_hook(module: nn.Module, _inputs, _outputs): function
945 def fw_hook(inc, h_module, input, output): function
1227 def fw_hook(inc, h_module, input, output): function
|
| /aosp_15_r20/external/pytorch/test/quantization/eager/ |
| H A D | test_quantize_eager_ptq.py | 732 def fw_hook(h_module, input, output): function
1457 def fw_hook(h_module, input, output): function
|
| H A D | test_fuse_eager.py | 398 def fw_hook(fused_module_class, h_module, input, output): function
|
| H A D | test_quantize_eager_qat.py | 528 def fw_hook(h_module, input, output): function
|
| /aosp_15_r20/external/pytorch/test/distributed/_tensor/ |
| H A D | test_dtensor_compile.py | 475 def fw_hook(module, inp, out): function
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_hooks.py | 775 def fw_hook(inp): function
|