Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_types.py12 from jit.test_module_interface import TestModuleInterface # noqa: F401
H A Dtest_misc.py11 from jit.test_module_interface import TestModuleInterface # noqa: F401
H A Dtest_module_interface.py44 class TestModuleInterface(JitTestCase): class
H A Dtest_tracer.py2731 class TestModuleInterface(nn.Module): class
2737 make_global(TestModuleInterface)
2749 def fn_takes_interface(x: TestModuleInterface): argument
/aosp_15_r20/external/pytorch/test/
H A Dtest_jit.py41 from jit.test_module_interface import TestModuleInterface # noqa: F401