-

#include "EndToEndTestImpl.hpp"
-#include "LogSoftmaxEndToEndTestImpl.hpp"
-#include <armnn/INetwork.hpp>
-#include <test/TestUtils.hpp>
-#include <boost/test/unit_test.hpp>
-

-

- - - - -

-Functions
void	LogSoftmaxEndToEndTest (const std::vector< armnn::BackendId > &defaultBackends)

-

Function Documentation

- -

◆ LogSoftmaxEndToEndTest()

- -

-

- - - - - - - - -

void LogSoftmaxEndToEndTest ( const std::vector< armnn::BackendId > & defaultBackends )

-

- -

Definition at line 76 of file LogSoftmaxEndToEndTestImpl.cpp.

- -

References armnn::Float32.

- -

Referenced by BOOST_AUTO_TEST_CASE().

-

 {
     using namespace armnn;
 
     const float beta = 10.0f; // non-default beta
     const int axis   = 3;     // positive axis
 
     const TensorShape inputShape{1, 1, 2, 4};
     TensorInfo inputTensorInfo(inputShape, DataType::Float32);
 
     const TensorShape outputShape{1, 1, 2, 4};
     TensorInfo outputTensorInfo(outputShape, DataType::Float32);
 
     std::vector<float> inputData = std::vector<float>({
         0.0f, -0.6f, 0.2f, 0.4f,
         0.3f, -0.2f, 1.0f, 0.1f
     });
 
     std::vector<float> expectedOutputData = std::vector<float>({
         -4.14297f, -10.14297f, -2.14297f, -0.14297f,
         -7.00104f, -12.00104f, -0.00104087f, -9.00104f
     });
 
     LogSoftmaxEndToEnd(defaultBackends,
                        inputTensorInfo,
                        outputTensorInfo,
                        inputData,
                        expectedOutputData,
                        beta,
                        axis);
 }

-
-
-
-
-

-

-Functions

Function Documentation

◆ LogSoftmaxEndToEndTest()