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, {})
|