| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_hooks_modules.py | 111 def pre_hook(self, input: Tuple[()]) -> None: function
127 def pre_hook(self, input: Tuple[()]) -> None: function
144 def pre_hook(self, input: Tuple[List[str], str]) -> Tuple[List[str], str]: function
205 def pre_hook(self, input: Tuple[str]) -> Tuple[str]: function
226 def pre_hook(self, input: Tuple[str]) -> Tuple[str]: function
249 def pre_hook(self, input: Tuple[str]) -> None: function
303 def pre_hook(self, input: Tuple[List[str], str]) -> Tuple[List[str], str]: function
366 def pre_hook(self, input: Tuple[str]) -> Tuple[str]: function
387 def pre_hook(self, input: Tuple[str]) -> Tuple[str]: function
406 def pre_hook(self, input: Tuple[str]) -> Tuple[str]: function
[all …]
|
| H A D | test_hooks.py | 180 def pre_hook(self, input: Tuple[str]) -> Tuple[str]: function
203 def pre_hook(self, input: Tuple[str]) -> Tuple[str]: function
232 def pre_hook(self, input: Tuple[str]) -> Tuple[str]: function
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/ |
| H A D | quantize.py | 155 def _register_activation_post_process_hook(module, pre_hook=False): argument
416 def remove_hooks(pre_hook=False): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | class_type.cpp | 41 for (const auto& pre_hook : forward_pre_hooks_) { in findForwardPreHook() local
195 const torch::jit::Function* pre_hook = forward_pre_hooks_[pre_hook_idx]; in checkForwardPreHookSchema() local
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_triton_heuristics.py | 133 def pre_hook(kwargs): function
|
| /aosp_15_r20/external/pytorch/torch/_export/ |
| H A D | wrappers.py | 83 def pre_hook(module, args, kwargs): function
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/api/ |
| H A D | compilation_unit.h | 240 for (const auto& pre_hook : cls->getForwardPreHooks()) { in _clear_python_cu() local
|
| H A D | module.cpp | 245 for (const auto& pre_hook : pre_forward_hooks) { in operator ()() local
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | compiled_autograd.py | 209 def pre_hook(self, inputs, hook_id): member in AutogradCompilerInstance
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/ |
| H A D | concrete_module_type.cpp | 285 void ConcreteModuleTypeBuilder::addForwardPreHook(py::object pre_hook) { in addForwardPreHook()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/ |
| H A D | python_print.cpp | 1462 for (const auto& pre_hook : forwardPreHooks) { in printClass() local
1530 for (auto& pre_hook : classType->getForwardPreHooks()) { in printClass() local
|
| H A D | import_source.cpp | 484 for (const auto& pre_hook : pre_hook_list) { in importClass() local
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_hooks.py | 682 def pre_hook(grad, *, k): function
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_optim.py | 1794 def pre_hook(opt: Optimizer, args: Tuple[Any], kwargs: Dict[Any, Any]): function
|
| H A D | test_autograd.py | 1386 def pre_hook(grad): function
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/python/ |
| H A D | script_init.cpp | 1243 for (auto& pre_hook : m.type()->getForwardPreHooks()) { in initJitScriptBindings() local
|