Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dutils.py2122 check_backward_hooks=False, argument
2133 and not check_backward_hooks
2138 if check_backward_hooks or check_all_hooks:
2156 check_backward_hooks=False, argument
2165 check_backward_hooks=check_backward_hooks,
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dnn_module.py407 mod, check_forward_hooks=True, check_backward_hooks=True