Searched refs:SimpleAxisSoftmaxTest (Results 1 – 3 of 3) sorted by relevance
| /external/armnn/src/backends/backendsCommon/test/layerTests/ |
| D | SoftmaxTestImpl.hpp | 21 LayerTestResult<float, 2> SimpleAxisSoftmaxTest(
|
| D | SoftmaxTestImpl.cpp | 324 LayerTestResult<float,2> SimpleAxisSoftmaxTest( in SimpleAxisSoftmaxTest() function
|
| /external/armnn/src/backends/reference/test/ |
| D | RefLayerTests.cpp | 696 ARMNN_AUTO_TEST_CASE_WITH_THF(Simple2dAxis0Softmax, SimpleAxisSoftmaxTest, 1.0f, 0)
697 ARMNN_AUTO_TEST_CASE_WITH_THF(Simple2dAxis1Softmax, SimpleAxisSoftmaxTest, 1.0f, 1)
699 ARMNN_AUTO_TEST_CASE_WITH_THF(Simple2dAxis0NegSoftmax, SimpleAxisSoftmaxTest, 1.0f, -2)
700 ARMNN_AUTO_TEST_CASE_WITH_THF(Simple2dAxis1NegSoftmax, SimpleAxisSoftmaxTest, 1.0f, -1)
|