Function ggml_axpy

ggml.c:4158–4190

}

struct ggml_tensor * ggml_axpy(
        struct ggml_context * ctx,
        struct ggml_tensor  * a,
        struct ggml_tensor  * b,
        struct ggml_tensor  * sparse_idx,
        // Under CPU + GPU hybrid inference:
        // When using GPU, this tensor is gpu_bucket to map the index back to the original tensor;
        // When using CPU, this tensor is gpu_index to indicate which row are offloaded to GPU.
        // Under full GPU/GPU inference, it is NULL.
        struct ggml_tensor  * hybrid_aux) {
    GGML_ASSERT(a != NULL && b != NULL);
    GGML_ASSERT(!ggml_is_transposed(a));
    bool is_node = false;

    if (a->grad || b->grad) {
        is_node = true;
    }

    const int64_t ne[4] = { a->ne[0], b->ne[1], b->ne[2], b->ne[3] };
    struct ggml_tensor * result = ggml_new_tensor(ctx, GGML_TYPE_F32, MAX(a->n_dims, b->n_dims), ne);

    result->op   = GGML_OP_AXPY;
    result->grad = is_node ? ggml_dup_tensor(ctx, result) : NULL;
    result->src[0] = a;
    result->src[1] = b;
    result->src[2] = sparse_idx;
    result->src[3] = hybrid_aux;

    int32_t params[] = { hybrid_aux ? 0 : 1 };
    ggml_set_op_params(result, params, sizeof(params));

    return result;
}

// ggml_out_prod