Searched defs:np_features (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| H A D | relu_op_test.py | 42 def _npRelu(self, np_features): argument
52 def _testRelu(self, np_features): argument
205 def _npRelu6(self, np_features): argument
218 def _testRelu6(self, np_features): argument
269 def _npLeakyRelu(self, np_features, alpha=0.1): argument
281 def _testLeakyRelu(self, np_features, alpha): argument
398 def _npElu(self, np_features): argument
409 def _testElu(self, np_features): argument
503 def _npSelu(self, np_features): argument
517 def _testSelu(self, np_features): argument
[all …]
|
| H A D | softsign_op_test.py | 29 def _npSoftsign(self, np_features): argument
32 def _testSoftsign(self, np_features, use_gpu=False): argument
|
| H A D | softplus_op_test.py | 30 def _npSoftplus(self, np_features): argument
35 def _testSoftplus(self, np_features, use_gpu=False): argument
|
| H A D | softmax_op_test.py | 56 np_features, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/distributions/ |
| H A D | util_test.py | 915 def _npSoftplus(self, np_features): argument
920 def _testSoftplus(self, np_features, use_gpu=False): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_mps.py | 1420 def _npRelu(self, np_features): argument
1430 def _testRelu(self, np_features, device): argument
1440 def _testReluInPlace(self, np_features, device): argument
1513 def _npLeakyRelu(self, np_features, negative_slope=0.1): argument
1657 def _testLeakyRelu(self, np_features, negative_slope, device): argument
|