Home
last modified time | relevance | path

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

/external/pytorch/test/
Dtest_transformers.py53 Tolerances = namedtuple('Tolerances', ['atol', 'rtol']) variable
3423 forw_tolerances: Optional[Tolerances] = None,
3424 grad_tolerances: Optional[Tolerances] = None,
/external/pytorch/test/inductor/
Dtest_flex_attention.py41 Tolerances = namedtuple("Tolerances", ["atol", "rtol"]) variable
Dtest_flex_decoding.py36 Tolerances = namedtuple("Tolerances", ["atol", "rtol"]) variable