Function sam3_find_state_tensor

sam3.cpp:14338–14380

*****************************************************************************/

static struct ggml_tensor * sam3_find_state_tensor(const sam3_state & state,
                                                   const std::string & tensor_name) {
    struct ggml_tensor* t = nullptr;

    if (tensor_name == "vit_output") {
        t = state.vit_output;
    } else if (tensor_name == "neck_det_0") {
        t = state.neck_det[0];
    } else if (tensor_name == "neck_det_1") {
        t = state.neck_det[1];
    } else if (tensor_name == "neck_det_2") {
        t = state.neck_det[2];
    } else if (tensor_name == "neck_det_3") {
        t = state.neck_det[3];
    } else if (tensor_name == "neck_trk_0") {
        t = state.neck_trk[0];
    } else if (tensor_name == "neck_trk_1") {
        t = state.neck_trk[1];
    } else if (tensor_name == "neck_trk_2") {
        t = state.neck_trk[2];
    } else if (tensor_name == "neck_trk_3") {
        t = state.neck_trk[3];
    } else if (tensor_name == "neck_det_pe_0") {
        t = state.neck_det_pe[0];
    } else if (tensor_name == "neck_det_pe_1") {
        t = state.neck_det_pe[1];
    } else if (tensor_name == "neck_det_pe_2") {
        t = state.neck_det_pe[2];
    } else if (tensor_name == "neck_det_pe_3") {
        t = state.neck_det_pe[3];
    } else {
        // Search by ggml name in the context
        if (state.ctx) {
            t = ggml_get_tensor(state.ctx, tensor_name.c_str());
        }
        // Also search PE context
        if (!t && state.pe_ctx) {
            t = ggml_get_tensor(state.pe_ctx, tensor_name.c_str());
        }
    }

    return t;
}

static bool sam3_fill_tensor_info(struct ggml_tensor * t, sam3_tensor_info & info) {
    if (!t) {