Function BatchToSpaceNDFp32Test

delegate/test/BatchSpaceTest.cpp:18–37

BatchToSpaceND Operator

// BatchToSpaceND Operator
void BatchToSpaceNDFp32Test()
{
    std::vector<int32_t> inputShape { 4, 1, 1, 1 };
    std::vector<int32_t> expectedOutputShape { 1, 2, 2, 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>> crops = {{0, 0}, {0, 0}};

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

void BatchToSpaceNDFp32BatchOneTest()
{