Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/
H A D_utils.py199 backward_hooks, argument
210 tensor._backward_hooks = backward_hooks
220 backward_hooks, argument
233 t._backward_hooks = backward_hooks
382 backward_hooks, argument
422 tensor._backward_hooks = backward_hooks
426 def _rebuild_parameter(data, requires_grad, backward_hooks): argument
431 param._backward_hooks = backward_hooks
436 def _rebuild_parameter_with_state(data, requires_grad, backward_hooks, state): argument
441 param._backward_hooks = backward_hooks
H A D_tensor.py264 backward_hooks: Dict[Any, Any] = OrderedDict()
360 backward_hooks,
486 backward_hooks,
/aosp_15_r20/external/pytorch/torch/autograd/
H A Dfunction.py279 def _register_hook(backward_hooks, hook): argument
280 if backward_hooks is None:
281 backward_hooks = OrderedDict()
282 handle = hooks.RemovableHandle(backward_hooks)
283 backward_hooks[handle.id] = hook
284 return backward_hooks, handle
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_variable.cpp440 Py_CLEAR(self->backward_hooks); in THPVariable_clear()
1278 if (self->backward_hooks) { in THPVariable_get_backwards_hooks()
1279 Py_INCREF(self->backward_hooks); in THPVariable_get_backwards_hooks()
1280 return self->backward_hooks; in THPVariable_get_backwards_hooks()
1299 Py_XDECREF(self->backward_hooks); in THPVariable_set_backwards_hooks()
1300 self->backward_hooks = obj; in THPVariable_set_backwards_hooks()
2233 Py_VISIT(var->backward_hooks); in THPVariable_subclass_traverse()
H A Dpython_variable.h23 PyObject* backward_hooks = nullptr; member
H A Dpython_cpp_function.cpp179 var->backward_hooks, THPVariable_Unpack(var).output_nr())); in THPCppFunction_register_hook_dict()
H A Dpython_function.cpp1404 new PyFunctionTensorPreHook(var->backward_hooks, tensor.output_nr())); in THPFunction__register_hook_dict()
/aosp_15_r20/external/pytorch/functorch/csrc/dim/
H A Dpython_variable_simple.h23 PyObject* backward_hooks = nullptr; member
/aosp_15_r20/external/pytorch/docs/source/
H A Dtorch.compiler_nn_module.rst32 occur when accessing backward_hooks dicts, which is probably avoiable with some work. Graph-breaks…