Function ReshapeInt16Test

delegate/test/ReshapeTest.cpp:205–231

}

void ReshapeInt16Test(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<int16_t> inputValues = { -5, 8, -10, 7,
                                         8, 12, -15, 2,
                                         3, -4, -1, -11 };

    std::vector<int16_t> expectedOutputValues = { -5, 8, -10, 7,
                                                  8, 12, -15, 2,
                                                  3, -4, -1, -11 };

    RedefineTest<int16_t>(tflite::BuiltinOperator_RESHAPE,
                          ::tflite::TensorType_INT16,
                          inputShape,
                          outputShape,
                          inputValues,
                          expectedOutputValues,
                          targetShape,
                          useOption,
                          backends,
                          2.5f,
                          0);
}

TEST_SUITE("ReshapeTests")
{