Function SplitVFp32Test

delegate/test/SplitTest.cpp:127–162

}

void SplitVFp32Test()
{
    std::vector<int32_t> axisShape { 1 };
    std::vector<int32_t> inputShape { 2, 4, 2, 2 };
    std::vector<int32_t> splitsShape { 2 };
    std::vector<int32_t> outputShape0 { 2, 3, 2, 2 };
    std::vector<int32_t> outputShape1 { 2, 1, 2, 2 };
    std::vector<std::vector<int32_t>> outputShapes{ outputShape0, outputShape1 };

    std::vector<int32_t> axisData { 1 };    // Axis
    std::vector<int32_t> splitsData { 3, 1 };  // Splits
    std::vector<float> inputValues { 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f,
                                     9.0f, 10.0f, 11.0f, 12.0f, 13.0f, 14.0f, 15.0f, 16.0f,
                                     17.0f, 18.0f, 19.0f, 20.0f, 21.0f, 22.0f, 23.0f, 24.0f,
                                     25.0f, 26.0f, 27.0f, 28.0f, 29.0f, 30.0f, 31.0f, 32.0f   }; // Input


    std::vector<float> expectedOutputValues0 { 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f,
                                               9.0f, 10.0f, 11.0f, 12.0f, 17.0f, 18.0f, 19.0f, 20.0f,
                                               21.0f, 22.0f, 23.0f, 24.0f, 25.0f, 26.0f, 27.0f, 28.0f };
    std::vector<float> expectedOutputValues1 { 13.0f, 14.0f, 15.0f, 16.0f, 29.0f, 30.0f, 31.0f, 32.0f };
    std::vector<std::vector<float>> expectedOutputValues{ expectedOutputValues0, expectedOutputValues1 };

    int32_t numSplits = 2;

    SplitVTest<float>(::tflite::TensorType_FLOAT32,
                      inputShape,
                      splitsShape,
                      axisShape,
                      outputShapes,
                      inputValues,
                      splitsData,
                      axisData,
                      expectedOutputValues,
                      numSplits);
}

void SplitVFp32ZeroDimTest()
{