Home
last modified time | relevance | path

Searched refs:setup_input_hook (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/utils/
H A Dhooks.py193 def setup_input_hook(self, args): member in BackwardHook
/aosp_15_r20/external/pytorch/torch/nn/modules/
H A Dmodule.py1788 args = bw_hook.setup_input_hook(args)