Searched defs:SoftmaxOpTest (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | activations_test.cc | 273 class SoftmaxOpTest : public SingleOpTest { class
1272 TEST_P(SoftmaxOpTest, Softmax4D) { in TEST_P() argument
1305 TEST_P(SoftmaxOpTest, Softmax4DUint8) { in TEST_P() argument
1344 TEST_P(SoftmaxOpTest, Softmax4DUint8Int16) { in TEST_P() argument
1385 TEST_P(SoftmaxOpTest, Softmax1DInt8) { in TEST_P() argument
1399 TEST_P(SoftmaxOpTest, Softmax1DInt16) { in TEST_P() argument
1414 TEST_P(SoftmaxOpTest, Softmax1DInt16ZeroElement) { in TEST_P() argument
1426 TEST_P(SoftmaxOpTest, Softmax2DInt16) { in TEST_P() argument
1469 TEST_P(SoftmaxOpTest, Softmax3DInt16) { in TEST_P() argument
1514 TEST_P(SoftmaxOpTest, Softmax4DInt16) { in TEST_P() argument
[all …]
|
| 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
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_grad_test.py | 38 class SoftmaxOpTest(test.TestCase): class
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 1346 TEST(SoftmaxOpTest, SimpleTest) { in TEST() argument
1362 TEST(SoftmaxOpTest, Beta2) { in TEST() argument
|