Function PreluFloatSimpleTest

delegate/test/PreluTest.cpp:12–36

namespace armnnDelegate {

void PreluFloatSimpleTest(bool isAlphaConst, bool isDynamicOutput = false)
{
    std::vector<int32_t> inputShape { 1, 2, 3 };
    std::vector<int32_t> alphaShape { 1 };
    std::vector<int32_t> outputShape { 1, 2, 3 };

    if (isDynamicOutput)
    {
        outputShape.clear();
    }

    std::vector<float> inputData = { -14.f, 2.f, 0.f, 1.f, -5.f, 14.f };
    std::vector<float> alphaData = { 0.5f };
    std::vector<float> expectedOutput = { -7.f, 2.f, 0.f, 1.f, -2.5f, 14.f };

    PreluTest(tflite::BuiltinOperator_PRELU,
              ::tflite::TensorType_FLOAT32,
              inputShape,
              alphaShape,
              outputShape,
              inputData,
              alphaData,
              expectedOutput,
              isAlphaConst);
}

TEST_SUITE("PreluTests")
{