Searched defs:SoftmaxOpTest (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | activations_test.cc | 280 class SoftmaxOpTest : public SingleOpTest { class
1354 TEST_P(SoftmaxOpTest, Softmax4D) { in TEST_P() argument
1387 TEST_P(SoftmaxOpTest, Softmax4DUint8) { in TEST_P() argument
1426 TEST_P(SoftmaxOpTest, Softmax4DUint8Int16) { in TEST_P() argument
1467 TEST_P(SoftmaxOpTest, Softmax1DInt8) { in TEST_P() argument
1481 TEST_P(SoftmaxOpTest, Softmax1DInt16) { in TEST_P() argument
1496 TEST_P(SoftmaxOpTest, Softmax1DInt16ZeroElement) { in TEST_P() argument
1508 TEST_P(SoftmaxOpTest, Softmax2DInt16) { in TEST_P() argument
1551 TEST_P(SoftmaxOpTest, Softmax3DInt16) { in TEST_P() argument
1596 TEST_P(SoftmaxOpTest, Softmax4DInt16) { in TEST_P() argument
[all …]
|
| H A D | softmax_test.cc | 63 TEST(SoftmaxOpTest, SimpleTest) { in TEST() argument
80 TEST(SoftmaxOpTest, CompareWithTFminiBetaEq1) { in TEST() argument
109 TEST(SoftmaxOpTest, CompareWithTFminiBetaNotEq1) { in TEST() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | nn_grad_test.py | 34 class SoftmaxOpTest(test.TestCase): class
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| H A D | nnapi_delegate_test.cc | 1423 TEST(SoftmaxOpTest, SimpleTest) { in TEST() argument
1439 TEST(SoftmaxOpTest, Beta2) { in TEST() argument
|