feat(webgl): update pool2d_avg_adaptive count params

JingyuanZhang · JingyuanZhang · commit e22f7898f36a · 2022-07-28T13:16:14.000+08:00
diff --git a/packages/paddlejs-backend-webgl/src/ops/shader/pool2d_avg_adaptive.ts b/packages/paddlejs-backend-webgl/src/ops/shader/pool2d_avg_adaptive.ts
@@ -18,7 +18,6 @@ function mainFunc(
         float res = 0.0;
         // 获取output的坐标
         ivec4 out_pos = getOutputTensorPos();
-        int count_pool = 0;
         int i = out_pos[2] * ${stride_v} - ${padTop};
         int j = out_pos[3] * ${stride_h} - ${padLeft};
         int hstart = int(floor(float(i) * float(${H}) / float(${ksize_y})));
@@ -29,9 +28,9 @@ function mainFunc(
             for (int fx = wstart; fx < wend; fx++) {
                 float curr = getValueFromTensorPos_origin(out_pos[0], out_pos[1], fy, fx);
                 res += curr;
-                count_pool++;
             }
         }
+        int count_pool = (hend - hstart) * (wend - wstart);
         res = res / float(count_pool);
         setOutput(res);
     }

Original file line number	Diff line number	Diff line change
`@@ -18,7 +18,6 @@ function mainFunc(`
`18`	`18`	`float res = 0.0;`
`19`	`19`	`// 获取output的坐标`
`20`	`20`	`ivec4 out_pos = getOutputTensorPos();`
`21`		`- int count_pool = 0;`
`22`	`21`	`int i = out_pos[2] * ${stride_v} - ${padTop};`
`23`	`22`	`int j = out_pos[3] * ${stride_h} - ${padLeft};`
`24`	`23`	`int hstart = int(floor(float(i) * float(${H}) / float(${ksize_y})));`
`@@ -29,9 +28,9 @@ function mainFunc(`
`29`	`28`	`for (int fx = wstart; fx < wend; fx++) {`
`30`	`29`	`float curr = getValueFromTensorPos_origin(out_pos[0], out_pos[1], fy, fx);`
`31`	`30`	`res += curr;`
`32`		`- count_pool++;`
`33`	`31`	`}`
`34`	`32`	`}`
	`33`	`+ int count_pool = (hend - hstart) * (wend - wstart);`
`35`	`34`	`res = res / float(count_pool);`
`36`	`35`	`setOutput(res);`
`37`	`36`	`}`