Searched refs:_test_activation_module_impl (Results 1 – 1 of 1) sorted by relevance
/aosp_15_r20/external/pytorch/test/quantization/core/ |
H A D | test_quantized_module.py | 1152 …def _test_activation_module_impl(self, name, float_module_class, quantized_module_class, extra_kwa… member in TestStaticQuantizedModule 1198 self._test_activation_module_impl("ELU", nn.ELU, nnq.ELU, {"alpha": 1.5}) 1201 …self._test_activation_module_impl("LeakyReLU", nn.LeakyReLU, nnq.LeakyReLU, {"negative_slope": 0.2… 1205 self._test_activation_module_impl("Sigmoid", nn.Sigmoid, nnq.Sigmoid, {}) 1223 self._test_activation_module_impl("Hardswish", nn.Hardswish, nnq.Hardswish, {})
|