Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/
H A Dtest_nn.py7908 def _test_module_empty_inputs(self, module, inputs): member in TestNNDeviceType
8738 self._test_module_empty_inputs(decoder_layer, [tgt, memory])
8749 self._test_module_empty_inputs(transformer_decoder, [tgt, memory])
8758 self._test_module_empty_inputs(transformer_model, [src, tgt])