Function SplitUint8Test

delegate/test/SplitTest.cpp:14–41

SPLIT Operator

// SPLIT Operator
void SplitUint8Test()
{
    std::vector<int32_t> axisShape { 1 };
    std::vector<int32_t> inputShape { 2, 2, 2, 2} ;
    std::vector<int32_t> outputShape0 { 2, 2, 2, 1 };
    std::vector<int32_t> outputShape1 { 2, 2, 2, 1 };
    std::vector<std::vector<int32_t>> outputShapes{ outputShape0, outputShape1 };

    std::vector<int32_t> axisData { 3 };  // Axis
    std::vector<uint8_t> inputValues { 1, 2, 3, 4, 5, 6, 7, 8,
                                       9, 10, 11, 12, 13, 14, 15, 16 }; // Input


    std::vector<uint8_t> expectedOutputValues0 { 1, 3, 5, 7, 9, 11, 13, 15 };
    std::vector<uint8_t> expectedOutputValues1 { 2, 4, 6, 8, 10, 12, 14, 16 };
    std::vector<std::vector<uint8_t>> expectedOutputValues{ expectedOutputValues0, expectedOutputValues1 };

    int32_t numSplits = 2;

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

void SplitFp32Test()
{