Method prepare

src/core/NEON/kernels/NEStackLayerKernel.cpp:192–228

}

void NEStackLayerKernel::prepare()
{
    // Prepare calculates the window at runtime, in case there is padding being added after configure()
    const ITensorInfo *input_info  = _input[0]->info();
    const int32_t      num_dims    = input_info->num_dimensions();
    const int32_t      num_tensors = _input.size();

    // Check if there are any paddings in the input tensors
    bool has_padding = false;
    for (const ITensor *in : _input)
    {
        if (has_holes(*in->info(), num_dims - 1))
        {
            has_padding = true;
            break;
        }
    }

    has_padding = has_padding || has_holes(*_output->info(), num_dims);

    Window win;
    if (!has_padding)
    {
        _stack_fn = memcpy_stack;

        // 2D execution window (X,Y): [Num_tensors, Dimensions >= axis]
        win.set(Window::DimX, Window::Dimension(0, num_tensors, 1));
        win.set(Window::DimY, Window::Dimension(0, input_info->tensor_shape().total_size_upper(_axis), 1));
    }
    else
    {
        _stack_fn = elementwise_stack;
        win       = calculate_max_window(*input_info);
    }

    INEKernel::configure(win);
}

void NEStackLayerKernel::run(const Window &window, const ThreadInfo &info)
{