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
|