Searched refs:_test_module_empty_inputs (Results 1 – 1 of 1) sorted by relevance
7908 def _test_module_empty_inputs(self, module, inputs): member in TestNNDeviceType8738 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])