Searched refs:_test_activation_function (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_quantized_op.py | 149 def _test_activation_function(self, X, fn_name, test_configs): member in TestQuantizedOps
276 self._test_activation_function(X, 'relu', relu_test_configs)
298 self._test_activation_function(X, 'relu6', relu6_test_configs)
315 self._test_activation_function(X, 'sigmoid', sigmoid_test_configs)
335 self._test_activation_function(X, 'sigmoid', sigmoid_test_configs)
352 self._test_activation_function(X, 'sigmoid', sigmoid_test_configs)
383 self._test_activation_function(X, 'hardsigmoid', hardsigmoid_test_configs)
402 self._test_activation_function(X, 'leaky_relu', leaky_relu_test_configs)
|