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
|