Searched refs:_test_module_empty_input (Results 1 – 6 of 6) sorted by relevance
/aosp_15_r20/external/pytorch/test/nn/ |
H A D | test_pooling.py | 35 _test_module_empty_input, 501 _test_module_empty_input(self, mod, inp, check_size=False) 505 _test_module_empty_input(self, mod, inp, check_size=False) 509 _test_module_empty_input(self, mod, inp, check_size=False) 552 _test_module_empty_input(self, mod, inp, check_size=False) 562 _test_module_empty_input(self, mod, inp, check_size=False) 614 _test_module_empty_input(self, mod, inp, check_size=False) 621 _test_module_empty_input(self, mod, inp, check_size=False) 629 _test_module_empty_input(self, mod, inp, check_size=False) 783 _test_module_empty_input(self, mod, inp, check_size=False) [all …]
|
H A D | test_convolution.py | 45 from torch.testing._internal.common_nn import _test_module_empty_input, NNTestCase 3004 _test_module_empty_input(self, mod, inp, check_size=False) 3012 _test_module_empty_input(self, mod, inp, check_size=False) 3020 _test_module_empty_input(self, mod, inp, check_size=False) 3031 _test_module_empty_input(self, mod, inp, check_size=False) 3034 _test_module_empty_input(self, mod, inp, check_size=False) 3041 _test_module_empty_input(self, mod, inp, check_size=False) 3044 _test_module_empty_input(self, mod, inp, check_size=False) 3051 _test_module_empty_input(self, mod, inp, check_size=False) 3054 _test_module_empty_input(self, mod, inp, check_size=False)
|
/aosp_15_r20/external/pytorch/test/xpu/ |
H A D | test_conv.py | 21 from torch.testing._internal.common_nn import _test_module_empty_input, NNTestCase 706 _test_module_empty_input(self, mod, inp, check_size=False) 714 _test_module_empty_input(self, mod, inp, check_size=False) 722 _test_module_empty_input(self, mod, inp, check_size=False) 733 _test_module_empty_input(self, mod, inp, check_size=False) 740 _test_module_empty_input(self, mod, inp, check_size=False) 747 _test_module_empty_input(self, mod, inp, check_size=False)
|
/aosp_15_r20/external/pytorch/test/ |
H A D | test_nn.py | 41 …eference, new_module_tests, single_batch_reference_fn, _test_bfloat16_ops, _test_module_empty_input 8411 _test_module_empty_input(self, mod, inp) 8414 _test_module_empty_input(self, mod, inp) 8561 _test_module_empty_input(self, mod, inp, check_size=False) 8704 … _test_module_empty_input(self, encoder_layer, input, check_size=False, inference=True) 8713 … _test_module_empty_input(self, encoder_layer, nt, check_size=False, inference=True) 8716 … _test_module_empty_input(self, encoder_layer, nt, check_size=False, inference=True) 8718 _test_module_empty_input(self, encoder_layer, input, check_size=False) 8728 _test_module_empty_input(self, transformer_encoder, input, check_size=False) 8767 _test_module_empty_input(self, mod, inp, check_size=False) [all …]
|
H A D | test_mps.py | 3937 def _test_module_empty_input(self, module, inp, check_size=True): member in TestMPS
|
/aosp_15_r20/external/pytorch/torch/testing/_internal/ |
H A D | common_nn.py | 3953 def _test_module_empty_input(test_case, module, inp, check_size=True, inference=False): function
|