Searched defs:hook_id (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | compiled_autograd.py | 196 def tensor_pre_hook(self, inputs, hook_id, i: int): argument
209 def pre_hook(self, inputs, hook_id): argument
222 def post_hook(self, outputs, inputs, hook_id): argument
236 def post_acc_grad_hook(self, input, hook_id): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | class_type.cpp | 140 const std::string& hook_id, in checkForwardHookInputArguments()
196 std::string hook_id = in checkForwardPreHookSchema() local
293 std::string hook_id = in checkForwardHookSchema() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/ |
| H A D | concrete_module_type.cpp | 328 for (const auto& hook_id : data_.forwardHooks_) { in dump() local
|