pool2d float padding fallback

2024-01-31 10:22:09 +08:00 · 2024-01-31 10:22:09 +08:00 · 66dd123b0f
commit 66dd123b0f
parent ca4ec6d867
2 changed files with 4 additions and 4 deletions
--- a/ggml.c
+++ b/ggml.c
@ -5569,8 +5569,8 @@ struct ggml_tensor * ggml_pool_2d(
        int                   k1,
        int                   s0,
        int                   s1,
-        int                   p0,
+        float                 p0,
-        int                   p1) {
+        float                 p1) {
    bool is_node = false;
--- a/ggml.h
+++ b/ggml.h
@ -1600,8 +1600,8 @@ extern "C" {
            int                   k1,
            int                   s0,
            int                   s1,
-            int                   p0,
+            float                 p0,
-            int                   p1);
+            float                 p1);
    // nearest interpolate
    // used in stable-diffusion