Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/
H A Dtest_matmul_cuda.py219 E5M2_MAX_POS = torch.finfo(torch.float8_e5m2fnuz).max variable
224 E5M2_MAX_POS = torch.finfo(torch.float8_e5m2).max variable
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_fp8.py26 E5M2_MAX_POS = torch.finfo(torch.float8_e5m2).max variable