Home
last modified time | relevance | path

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

/external/pytorch/test/
Dtest_nestedtensor.py6790 def test_dummy_mha_with_nt(self, device, use_legacy_api): argument
6802 def __init__(self, use_legacy_api) -> None: argument
6806 self.use_legacy_api = use_legacy_api
6810 if self.use_legacy_api:
6843 if self.use_legacy_api:
6854 m = mha(use_legacy_api)
6872 m_eager = mha(use_legacy_api)