Function SpaceToBatchNDFp32Test

delegate/test/BatchSpaceTest.cpp:82–101

SpaceToBatchND Operator

// SpaceToBatchND Operator
void SpaceToBatchNDFp32Test()
{
    std::vector<int32_t> inputShape { 1, 2, 2, 1 };
    std::vector<int32_t> expectedOutputShape { 4, 1, 1, 1 };

    std::vector<float> inputValues { 1.0f, 2.0f, 3.0f, 4.0f };
    std::vector<float> expectedOutputValues { 1.0f, 2.0f, 3.0f, 4.0f };

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

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

void SpaceToBatchNDFp32PaddingTest()
{