Function ReshapeFlattenTest

delegate/test/ReshapeTest.cpp:97–121

}

void ReshapeFlattenTest(bool useOption = true, const std::vector<armnn::BackendId>& backends = {})
{
    // Set input data
    std::vector<int32_t> inputShape { 1, 3, 4, 1 };
    std::vector<int32_t> outputShape { 6, 2 };
    std::vector<int32_t> targetShape { -1, 2 };

    std::vector<float> inputValues = { -5.0f, 8.0f, -10.0f, 7.0f,
                                       8.0f, 12.0f, -15.0f, 2.0f,
                                       3.0f, -4.0f, -1.0f, -11.0f };

    std::vector<float> expectedOutputValues = { -5.0f, 8.0f, -10.0f, 7.0f,
                                                8.0f, 12.0f, -15.0f, 2.0f,
                                                3.0f, -4.0f, -1.0f, -11.0f };

    RedefineTest<float>(tflite::BuiltinOperator_RESHAPE,
                        ::tflite::TensorType_FLOAT32,
                        inputShape,
                        outputShape,
                        inputValues,
                        expectedOutputValues,
                        targetShape,
                        useOption,
                        backends);
}

void ReshapeFlattenAllTest(bool useOption = true, const std::vector<armnn::BackendId>& backends = {})
{