Function SpaceToBatchNDFp32PaddingTest

delegate/test/BatchSpaceTest.cpp:103–129

}

void SpaceToBatchNDFp32PaddingTest()
{
    std::vector<int32_t> inputShape { 2, 2, 4, 1 };
    std::vector<int32_t> expectedOutputShape { 8, 1, 3, 1 };

    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 };

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

    std::vector<unsigned int> blockShape({2, 2});
    std::vector<std::pair<unsigned int, unsigned int>> padding = {{0, 0}, {2, 0}};

    BatchSpaceTest<float>(tflite::BuiltinOperator_SPACE_TO_BATCH_ND,
                          ::tflite::TensorType_FLOAT32,
                          inputShape,
                          expectedOutputShape,
                          inputValues,
                          blockShape,
                          padding,
                          expectedOutputValues);
}

void SpaceToBatchNDUint8Test()
{