Home
last modified time | relevance | path

Searched defs:check_nan (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/autograd/
H A Danomaly_mode.py77 def __init__(self, check_nan=True) -> None: # noqa: D107 argument
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Danomaly_mode.cpp24 DetectAnomalyGuard::DetectAnomalyGuard(bool check_nan) { in DetectAnomalyGuard()