Home
last modified time | relevance | path

Searched defs:MockModule (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_activation_checkpointing.py310 class MockModule(torch.nn.Module): class
339 class MockModule(torch.nn.Module): class
426 class MockModule(torch.nn.Module): class
1123 class MockModule(torch.nn.Module): class
1147 class MockModule(torch.nn.Module): class
1247 class MockModule(torch.nn.Module): class
H A Dtest_modules.py1669 class MockModule(torch.nn.Module): class
1883 class MockModule(torch.nn.Module): class
1910 class MockModule(torch.nn.Module): class
1950 class MockModule(torch.nn.Module): class
2016 class MockModule(torch.nn.Module): class
2054 class MockModule(torch.nn.Module): class
2861 class MockModule(torch.nn.Module): class
H A Dtest_higher_order_ops.py2197 class MockModule(torch.nn.Module): class
2269 class MockModule(torch.nn.Module): class
3783 class MockModule(torch.nn.Module): class
6412 class MockModule(torch.nn.Module): class
H A Dtest_repros.py904 class MockModule(torch.nn.Module): class
2445 class MockModule(torch.nn.Module): class
H A Dtest_misc.py3625 class MockModule(torch.nn.Module): class
5921 class MockModule(torch.nn.Module): class
5949 class MockModule(torch.nn.Module): class
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dsequential.cpp218 struct MockModule : torch::nn::Module { in TEST_F() struct
219 explicit MockModule(int value) : expected(value) {} in TEST_F() argument
220 int expected; in TEST_F()
221 int forward(int value) { in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/python/util/
H A Dmodule_wrapper_test.py29 class MockModule(types.ModuleType): class
H A Dtf_export_test.py44 class MockModule(object): class in ValidateExportTest
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_minifier.py88 class MockModule(torch.nn.Module): class
H A Dtest_aotdispatch.py5069 class MockModule(torch.nn.Module): class
5804 class MockModule(torch.nn.Module): class
5833 class MockModule(torch.nn.Module): class
5896 class MockModule(torch.nn.Module): class
5915 class MockModule(torch.nn.Module): class
5956 class MockModule(torch.nn.Module): class
6002 class MockModule(torch.nn.Module): class
/aosp_15_r20/external/pytorch/test/
H A Dtest_stateless.py17 class MockModule(torch.nn.Module): class
H A Dtest_fx.py3289 class MockModule(torch.nn.Module): class
/aosp_15_r20/external/pytorch/test/higher_order_ops/
H A Dtest_with_effects.py364 class MockModule(torch.nn.Module): class
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_dynamo_distributed.py1432 class MockModule(torch.nn.Module): class
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Ddistributed_test.py10095 class MockModule(torch.nn.Module): class