Searched refs:assert_ref_equals_params (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_sdpa.py | 22 def assert_ref_equals_params(self, actual, expected): member in TestSDPA
42 self.assert_ref_equals_params(o, SDPAParams(q, k, v, m, 0.1, True, False))
61 self.assert_ref_equals_params(o, SDPAParams(q, k, v, m, 0.1, True, False))
99 self.assert_ref_equals_params(o, expected)
|