Searched refs:NewModuleTest (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_cpp_api_parity.py | 42 (sample_module.module_tests, common_nn.NewModuleTest),
43 (sample_functional.functional_tests, common_nn.NewModuleTest),
44 (common_nn.module_tests, common_nn.NewModuleTest),
45 (common_nn.new_module_tests, common_nn.NewModuleTest),
|
| H A D | test_nn.py | 39 from torch.testing._internal.common_nn import NNTestCase, NewModuleTest, CriterionTest, \
7281 test = NewModuleTest(**test_params)
7297 test = NewModuleTest(**test_params)
7348 test = NewModuleTest(**test_params)
7385 add_test(NewModuleTest(
7392 add_test(NewModuleTest(
7399 add_test(NewModuleTest(
7408 add_test(NewModuleTest(
7416 add_test(NewModuleTest(
7425 add_test(NewModuleTest(
[all …]
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_nn.py | 3613 class NewModuleTest(InputVariableMixin, ModuleTest): # type: ignore[misc] class
|