Function SplitFp32Test

delegate/test/SplitTest.cpp:43–70

}

void SplitFp32Test()
{
    std::vector<int32_t> axisShape { 1 };
    std::vector<int32_t> inputShape { 2, 2, 2, 2 };
    std::vector<int32_t> outputShape0 { 2, 1, 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<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  }; // Input


    std::vector<float> expectedOutputValues0 { 1.0f, 2.0f, 3.0f, 4.0f, 9.0f, 10.0f, 11.0f, 12.0f };
    std::vector<float> expectedOutputValues1 { 5.0f, 6.0f, 7.0f, 8.0f, 13.0f, 14.0f, 15.0f, 16.0f };
    std::vector<std::vector<float>> expectedOutputValues{ expectedOutputValues0, expectedOutputValues1 };

    int32_t numSplits = 2;

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

// SPLIT Test Suite
TEST_SUITE("SPLITTests")