feat(paddlejs-backend-webgl): add hard_swish、nearest_interp op

Author: changy1105

packages/paddlejs-backend-webgl/src/ops/index.ts

@@ -42,6 +42,9 @@ import connect from './shader/connect';
 import squeeze2 from './shader/squeeze2';
 import pad3d from './shader/pad3d';
 import reduce_mean from './shader/reduce_mean';
+import hard_swish from './shader/hard_swish';
+import nearest_interp from './shader/nearest_interp';
+import nearest_interp_v2 from './shader/nearest_interp_v2';
 
 const ops = {
     arg_max,
@@ -79,6 +82,9 @@ const ops = {
     where,
     connect,
     reduce_mean,
+    hard_swish,
+    nearest_interp,
+    nearest_interp_v2,
     prelu: dynamic('prelu'),
     relu6: dynamic('relu6'),
     leakyRelu: dynamic('leakyRelu'),

packages/paddlejs-backend-webgl/src/ops/shader/hard_swish.ts

@@ -0,0 +1,33 @@
+/**
+ * @file hard_swish
+ */
+
+function mainFunc(
+    {},
+    {
+        offset = 3.0,
+        scale = 6.0,
+        threshold = 6.0
+    }
+) {
+    return `
+    void main(void) {
+        // 输出数据
+        ivec4 oPos = getOutputTensorPos();
+        float o = getValueFromTensorPos_origin(oPos.r, oPos.g, oPos.b, oPos.a);
+        float res = o * min(max(0.0, o + float(${offset})), float(${threshold})) / float(${scale});
+        setOutput(res);
+    }
+    `;
+}
+export default {
+    mainFunc,
+    params: [
+        'offset',
+        'scale',
+        'threshold'
+    ],
+    textureFuncConf: {
+        origin: ['getValueFromTensorPos']
+    }
+};

packages/paddlejs-backend-webgl/src/ops/shader/nearest_interp.ts

@@ -0,0 +1,50 @@
+/**
+ * @file nearest_interp
+ */
+
+function mainFunc(
+    { origin, out },
+    { align_corners }
+) {
+
+    return `
+    // start函数
+    int getData(float n, float scale, bool align_corners) {
+        float m = align_corners ? (n / scale + 0.5) : (n / scale);
+        return int(floor(m));
+    }
+
+    void main(void) {
+        // 输出数据
+        ivec4 oPos = getOutputTensorPos();
+        
+        float scale_x = 0.0;
+        float scale_y = 0.0;
+        if (${align_corners}) {
+            scale_x = float(${out.width_shape} -1) / float(${origin.width_shape} - 1);
+            scale_y = float(${out.height_shape} - 1) / float(${origin.height_shape} - 1);
+        }
+        else {
+            scale_x = float(${out.width_shape}) / float(${origin.width_shape});
+            scale_y = float(${out.height_shape}) / float(${origin.height_shape});
+        }
+    
+        int vx = getData(float(oPos.a), scale_x, ${align_corners});
+        int vy = getData(float(oPos.b), scale_y, ${align_corners});
+        
+        float o = getValueFromTensorPos_origin(oPos.r, oPos.g, vy, vx);
+        setOutput(float(o));
+}
+    `;
+}
+
+export default {
+    mainFunc,
+    params: [
+        'align_corners'
+    ],
+    textureFuncConf: {
+        origin: ['getValueFromTensorPos']
+    },
+    commonFuncConf: ['transferFromNHWCtoNCHW']
+};

packages/paddlejs-backend-webgl/src/ops/shader/nearest_interp_v2.ts

@@ -0,0 +1,7 @@
+/**
+ * @file nearest_interp_v2
+ */
+
+import nearest_interp from './nearest_interp';
+
+export default nearest_interp;

packages/paddlejs-backend-webgl/test/op/data/hard_swish.json

@@ -0,0 +1,29 @@
+{
+    "ops": [
+        {
+            "attrs": {
+                "offset": 3,
+                "scale": 6,
+                "threshold": 6
+            },
+            "inputs": {
+                "X": ["hard_swish.tmp_0"]
+            },
+            "outputs": {
+                "Out": ["hard_swish.tmp_1"]
+            },
+            "type": "hard_swish"
+        }
+    ],
+    "vars": [
+        {
+            "data": [1, 2, 3, 4],
+            "name": "hard_swish.tmp_0",
+            "shape": [1, 4]
+        },
+        {
+            "name": "hard_swish.tmp_1",
+            "shape": [1, 4]
+        }
+    ]
+}

packages/paddlejs-backend-webgl/test/op/data/nearest_interp.json

@@ -0,0 +1,27 @@
+{
+    "ops": [
+        {
+            "attrs": {
+                "align_corners": false
+            },
+            "inputs": {
+                "X": ["nearest_interp.tmp_0"]
+            },
+            "outputs": {
+                "Out": ["nearest_interp.tmp_1"]
+            },
+            "type": "nearest_interp"
+        }
+    ],
+    "vars": [
+        {
+            "data": [2, 3, 6, 10],
+            "name": "nearest_interp.tmp_0",
+            "shape": [1, 4]
+        },
+        {
+            "name": "nearest_interp.tmp_1",
+            "shape": [1, 4]
+        }
+    ]
+}