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)
|