Searched refs:_test_rnn_impl (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/quantization/fx/ |
| H A D | test_quantize_fx.py | 8713 def _test_rnn_impl(self, qconfigs, M, module_type_strs, module_types, sample_input): member in TestQuantizeFxOps
8746 … self._test_rnn_impl(qconfigs, RNNCellDynamicModel, module_type_strs, module_types, sample_input)
8759 self._test_rnn_impl(qconfigs, RNNDynamicModel, module_type_strs, module_types, sample_input)
|