Method run

src/core/NEON/kernels/NEBatchToSpaceLayerKernel.cpp:172–249

}

void NEBatchToSpaceLayerKernel::run(const Window &window, const ThreadInfo &info)
{
    ARM_COMPUTE_UNUSED(info);
    ARM_COMPUTE_ERROR_ON_UNCONFIGURED_KERNEL(this);
    ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW(ICPPKernel::window(), window);

    if (_block_shape != nullptr)
    {
        // Retrieve the block shapes dynamically
        _block_shape_x = *(reinterpret_cast<const int *>(_block_shape->ptr_to_element(0)));
        _block_shape_y = *(reinterpret_cast<const int *>(_block_shape->ptr_to_element(1)));
    }

    const int batch_size   = _output->info()->dimension(3);
    const int element_size = _output->info()->element_size();

    Window slice_out = window.first_slice_window_3D();

    int batch_id = 0;
    // Main loop for NCHW and NHWC
    if (_data_layout == DataLayout::NCHW)
    {
        do
        {
            Iterator out(_output, slice_out);
            execute_window_loop(
                slice_out,
                [&](const Coordinates &id)
                {
                    const int x = id.x();
                    const int y = id.y();
                    const int z = id.z();
                    // Translate x, y to uncropped version
                    const int x_c = x + _crop_info.left;
                    const int y_c = y + _crop_info.top;

                    const int in_batch =
                        batch_id + ((x_c % _block_shape_x) + (y_c % _block_shape_y) * _block_shape_x) * batch_size;
                    const int   in_x = x_c / _block_shape_x;
                    const int   in_y = y_c / _block_shape_y;
                    Coordinates input_coords{in_x, in_y, z, in_batch};
                    memcpy(out.ptr(), _input->ptr_to_element(input_coords), element_size);
                },
                out);
            ++batch_id;
        } while (window.slide_window_slice_3D(slice_out));
    }
    else
    {
        // For NHWC we can perform a block copy on the Channel (first) dimension. Thus we do not need to iterate over this dimension
        slice_out.set(0U, Window::Dimension(0U, 1U, 1U));
        do
        {
            Iterator out(_output, slice_out);
            execute_window_loop(
                slice_out,
                [&](const Coordinates &id)
                {