Function SoftmaxTestCase

delegate/test/SoftmaxTestHelper.hpp:141–156

Convenience function to run softmax and log-softmax test cases \param operatorCode tflite::BuiltinOperator_SOFTMAX or tflite::BuiltinOperator_LOG_SOFTMAX \param backends armnn backends to target \param beta multiplicative parameter to the softmax function \param expectedOutput to be checked against transformed input

/// \param beta multiplicative parameter to the softmax function
/// \param expectedOutput to be checked against transformed input
void SoftmaxTestCase(tflite::BuiltinOperator operatorCode, float beta,
                     std::vector<float> expectedOutput, const std::vector<armnn::BackendId> backends = {})
{
    std::vector<float> input = {
        1.0, 2.5, 3.0, 4.5, 5.0,
        -1.0, -2.5, -3.0, -4.5, -5.0};
    std::vector<int32_t> shape = {2, 5};

    SoftmaxTest(operatorCode,
                tflite::TensorType_FLOAT32,
                shape,
                input,
                expectedOutput,
                backends,
                beta);
}

} // anonymous namespace