Searched refs:use_legacy_api (Results 1 – 1 of 1) sorted by relevance
6790 def test_dummy_mha_with_nt(self, device, use_legacy_api): argument6802 def __init__(self, use_legacy_api) -> None: argument6806 self.use_legacy_api = use_legacy_api6810 if self.use_legacy_api:6843 if self.use_legacy_api:6854 m = mha(use_legacy_api)6872 m_eager = mha(use_legacy_api)