Function edgetam_window_unpartition_cpu

sam3.cpp:5580–5604

Inverse of window partition: [D, 16, 256] → [D, H, W] spatial layout. Rearranges 256 windows of 16 tokens each back into [D, H*W] feature map. Output has ggml layout [D, W, H] (ne[0]=D, ne[1]=W, ne[2]=H).

// Rearranges 256 windows of 16 tokens each back into [D, H*W] feature map.
// Output has ggml layout [D, W, H] (ne[0]=D, ne[1]=W, ne[2]=H).
static void edgetam_window_unpartition_cpu(
        const float* src, int D, int H, int W,
        int window_size, float* dst) {
    const int nw_h = H / window_size;
    const int nw_w = W / window_size;
    const int ws2  = window_size * window_size;

    for (int wh = 0; wh < nw_h; ++wh) {
        for (int ww = 0; ww < nw_w; ++ww) {
            int win_idx = wh * nw_w + ww;
            for (int ph = 0; ph < window_size; ++ph) {
                for (int pw = 0; pw < window_size; ++pw) {
                    int tok_idx = ph * window_size + pw;
                    int dst_h   = wh * window_size + ph;
                    int dst_w   = ww * window_size + pw;
                    int dst_pos = dst_w + dst_h * W;

                    const float* s = src + win_idx * (D * ws2) + tok_idx * D;
                    float* d       = dst + dst_pos * D;
                    memcpy(d, s, D * sizeof(float));
                }
            }
        }
    }
}

static bool edgetam_perceiver_forward(
        const sam3_model& model,