| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_hooks_modules.py | 114 def forward_hook(self, input: Tuple[()], output: None): function
130 def forward_hook(self, input: Tuple[()], output: None): function
149 def forward_hook(self, input: Tuple[List[str], str], output: Tuple[List[str], str]): function
210 def forward_hook(self, input: Tuple[str], output: str): function
231 def forward_hook(self, input: Tuple[str], output: str): function
253 def forward_hook(self, input: Tuple[str], output: str): function
308 def forward_hook(self, input: Tuple[List[str], str], output: Tuple[List[str], str]): function
371 def forward_hook(self, input: Tuple[str], output: str): function
391 def forward_hook(self, input: Tuple[str], output: str): function
411 def forward_hook(self, input: Tuple[str], output: str): function
[all …]
|
| H A D | test_hooks.py | 183 def forward_hook(self, input: Tuple[str], output: str): function
206 def forward_hook(self, input: Tuple[str], output: str): function
|
| /aosp_15_r20/external/pytorch/torch/distributed/_composable/ |
| H A D | checkpoint_activation.py | 108 def forward_hook(module: nn.Module, inputs: Tuple[Any, ...], output: Any) -> Any: function
|
| /aosp_15_r20/external/pytorch/test/nn/ |
| H A D | test_module_hooks.py | 56 def forward_hook( function
1067 def forward_hook(m, input, output): function
1636 def forward_hook(m, input, output): function
|
| /aosp_15_r20/external/pytorch/test/distributed/_composable/ |
| H A D | test_contract.py | 35 def forward_hook( function
|
| /aosp_15_r20/external/pytorch/test/higher_order_ops/ |
| H A D | test_with_effects.py | 375 def forward_hook( function
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_modules.py | 2175 def forward_hook( function
2222 def forward_hook( function
2291 def forward_hook( function
|
| /aosp_15_r20/external/pytorch/test/export/ |
| H A D | test_export.py | 5075 def forward_hook(module: torch.nn.Module, inputs, output) -> torch.Tensor: function
|