Searched refs:test_output_match (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/test/onnx/ | ||
| D | test_op_consistency.py | 281 def test_output_match(self, device: str, dtype: torch.dtype, op): member in TestOnnxModelOutputConsistency |
| /external/pytorch/test/ | ||
| D | test_mps.py | 12111 def test_output_match(self, device, dtype, op): member in TestConsistency |