Home
last modified time | relevance | path

Searched defs:check_batched_forward_grad (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/autograd/
H A Dgradcheck.py2065 check_batched_forward_grad, argument
/aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/
H A Dcore.py816 check_batched_forward_grad: bool = None variable in OpInfo
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_utils.py5096 check_batched_grad=None, check_batched_forward_grad=False): argument
5210 check_batched_grad=None, check_batched_forward_grad=False): argument