| 2385 | // ggml_sum_rows |
| 2386 | |
| 2387 | struct ggml_tensor * ggml_sum_rows( |
| 2388 | struct ggml_context * ctx, |
| 2389 | struct ggml_tensor * a) { |
| 2390 | int64_t ne[GGML_MAX_DIMS] = { 1 }; |
| 2391 | for (int i = 1; i < GGML_MAX_DIMS; ++i) { |
| 2392 | ne[i] = a->ne[i]; |
| 2393 | } |
| 2394 | |
| 2395 | struct ggml_tensor * result = ggml_new_tensor(ctx, a->type, GGML_MAX_DIMS, ne); |
| 2396 | |
| 2397 | result->op = GGML_OP_SUM_ROWS; |
| 2398 | result->src[0] = a; |
| 2399 | |
| 2400 | return result; |
| 2401 | } |
| 2402 | |
| 2403 | // ggml_cumsum |
| 2404 |