Searched refs:reference_in_float (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_foreach.py | 326 reference_in_float=False,
389 fn, gen_args(op), reference_in_float=False, check_lowp=False
440 reference_in_float=False,
465 reference_in_float=False,
|
| H A D | test_pattern_matcher.py | 45 reference_in_float=False, argument
49 if reference_in_float:
60 torch.testing.assert_close(actual, expected, check_dtype=not reference_in_float)
1396 self.common(mul_softmax, (x, scale), 1, 4, reference_in_float=True)
1397 self.common(mul_softmax, (scale, x), 1, 4, reference_in_float=True)
1398 self.common(div_softmax, (x, inv_scale), 1, 4, reference_in_float=True)
|
| H A D | test_torchinductor.py | 380 reference_in_float=True, argument
393 if reference_in_float and exact_dtype:
410 if reference_in_float:
475 if reference_in_float and exact_dtype:
482 if reference_in_float:
552 if reference_in_float:
584 reference_in_float=True, argument
610 reference_in_float=reference_in_float,
642 reference_in_float=reference_in_float,
1655 reference_in_float=True,
[all …]
|