Merge pull request PaddlePaddle#395 from JingyuanZhang/master

feat(webgl): support pool2d adaptive and adapt split num attr
zhicheng99 · Mar 8, 2022 · 1555bcb · 1555bcb
2 parents caef32b + 855c9b3
commit 1555bcb
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 3 deletions.
diff --git a/packages/paddlejs-backend-webgl/package-lock.json b/packages/paddlejs-backend-webgl/package-lock.json
diff --git a/packages/paddlejs-backend-webgl/package.json b/packages/paddlejs-backend-webgl/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@paddlejs/paddlejs-backend-webgl",
-  "version": "1.1.15",
+  "version": "1.1.17",
   "description": "",
   "main": "lib/index",
   "scripts": {

diff --git a/packages/paddlejs-core/package.json b/packages/paddlejs-core/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@paddlejs/paddlejs-core",
-  "version": "2.1.15",
+  "version": "2.1.16",
   "description": "",
   "main": "lib/index",
   "scripts": {

diff --git a/packages/paddlejs-core/src/opFactory/opBehaviors.ts b/packages/paddlejs-core/src/opFactory/opBehaviors.ts
@@ -211,6 +211,10 @@ const behaviors : Behaviors = {
         // 保存 输入 tensor 对应dim 的长度
         this.processedAttrs.inputs_dim = shape[axis];
         this.processedAttrs.dim = axis;
+        if (this.processedAttrs.num === 0) {
+            this.processedAttrs.num = Object.values(this.tensorDataMap)
+                .filter(item => item.tensorName === 'out').length || 1;
+        }
     },
 
     processElementwiseAxis() {