Home
last modified time | relevance | path

Searched refs:assert_ref_equals_params (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_sdpa.py22 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)