Function ref_reorg_common

executor/operator/ref/kernel/reorg/ref_reorg_kernel.h:48–78

typedef int (*ref_reorg_kernel_t)(const void* in_data, void* out_data, ref_reorg_param* param);

static int ref_reorg_common(const char* in_data, char* out_data, ref_reorg_param* param, int element_size)
{
    int w = param->w;
    int h = param->h;
    int c = param->c;
    int batch = param->batch;
    int stride = param->stride;

    int out_c = c / (stride * stride);

    for(int b = 0; b < batch; ++b)
    {
        for(int k = 0; k < c; ++k)
        {
            for(int j = 0; j < h; ++j)
            {
                for(int i = 0; i < w; ++i)
                {
                    int in_index = i + w * (j + h * (k + c * b));
                    int c2 = k % out_c;
                    int offset = k / out_c;
                    int w2 = i * stride + offset % stride;
                    int h2 = j * stride + offset / stride;
                    int out_index = w2 + w * stride * (h2 + h * stride * (c2 + out_c * b));
                    memcpy(out_data + in_index * element_size, in_data + out_index * element_size, element_size);
                }
            }
        }
    }
    return 0;
}

#ifdef CONFIG_KERNEL_FP32
#include "ref_reorg_fp32.c"