Home
last modified time | relevance | path

Searched refs:set_detect_anomaly (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/autograd/
H A Danomaly_mode.py95 class set_detect_anomaly: class
H A D__init__.py20 from .anomaly_mode import detect_anomaly, set_detect_anomaly
/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Dnumeric_utils.py200 with torch.autograd.set_detect_anomaly(True):
/aosp_15_r20/external/pytorch/docs/source/
H A Dautograd.rst303 .. autoclass:: set_detect_anomaly
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_vmap.py5256 with torch.autograd.set_detect_anomaly(True):