Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_quantize_fx.py8713 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)