Home
last modified time | relevance | path

Searched refs:_test_module_empty_input (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/test/nn/
H A Dtest_pooling.py35 _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 Dtest_convolution.py45 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 Dtest_conv.py21 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 Dtest_nn.py41 …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 Dtest_mps.py3937 def _test_module_empty_input(self, module, inp, check_size=True): member in TestMPS
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_nn.py3953 def _test_module_empty_input(test_case, module, inp, check_size=True, inference=False): function