Searched refs:run_test_with_call (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_flex_decoding.py | 310 def run_test_with_call( member in TestFlexDecoding
834 self.run_test_with_call(attention, sdpa_attention, Q_H=16, KV_H=16, Q_S=8)
853 self.run_test_with_call(attention, sdpa_attention, Q_H=16, KV_H=16, Q_S=8)
868 self.run_test_with_call(attention, sdpa_attention, Q_H=16, KV_H=16, Q_S=8)
|