| 3243 | // ggml_out_prod |
| 3244 | |
| 3245 | static inline bool ggml_can_out_prod(const struct ggml_tensor * t0, const struct ggml_tensor * t1) { |
| 3246 | static_assert(GGML_MAX_DIMS == 4, "GGML_MAX_DIMS is not 4 - update this function"); |
| 3247 | |
| 3248 | return (t0->ne[1] == t1->ne[1]) && |
| 3249 | (t1->ne[2]%t0->ne[2] == 0) && // verify t0 is broadcastable |
| 3250 | (t1->ne[3]%t0->ne[3] == 0); |
| 3251 | } |
| 3252 | |
| 3253 | struct ggml_tensor * ggml_out_prod( |
| 3254 | struct ggml_context * ctx, |