Home
last modified time | relevance | path

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

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