webgl: operator Tanh (#62)

* Add tanh ops for webgl backend * A faster tanh ops for webgl backend
microsoft · Jan 9, 2019 · 653fc9e · 653fc9e
1 parent 35ed071
commit 653fc9e
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 2 deletions.
diff --git a/lib/backends/webgl/ops/unary-op.ts b/lib/backends/webgl/ops/unary-op.ts
@@ -149,6 +149,22 @@ export function glslSqrt(): GlslValueFunction {
 export function glslTan(): GlslValueFunction {
   return glslBuiltinUnary('tan');
 }
+export function glslTanh(): GlslValueFunction {
+  const name = `tanh_`;
+  const body = `
+  float ${name}(float a) {
+    a = clamp(a, -10., 10.);
+    a = exp(2.*a);
+    return (a - 1.) / (a + 1.);
+  }
+  vec4 ${name}(vec4 v) {
+    v = clamp(v, -10., 10.);
+    v = exp(2.*v);
+    return (v - 1.) / (v + 1.);
+  }
+  `;
+  return {body, name, type: FunctionType.ValueBased};
+}
 function glslBuiltinUnary(fname: string): GlslValueFunction {
   const name = `${fname}_`;
   const body = `

diff --git a/lib/backends/webgl/session-handler.ts b/lib/backends/webgl/session-handler.ts
@@ -198,6 +198,8 @@ export class WebGLSessionHandler implements SessionHandler {
         return new WebGLSlice();
       case 'Tan':
         return new unaryOps.WebGLUnaryOp(FLOAT_TYPES, unaryOps.glslTan());
+      case 'Tanh':
+        return new unaryOps.WebGLUnaryOp(FLOAT_TYPES, unaryOps.glslTanh());
       case 'Transpose':
         return new WebGLTranspose();
       case 'Tile':

diff --git a/test/unittest-whitelist.jsonc b/test/unittest-whitelist.jsonc
@@ -403,8 +403,8 @@
       // "test_slice_start_out_of_bounds", // tensor shape of 0
       "test_tan_example",
       "test_tan",
-      // "test_tanh_example",
-      // "test_tanh",
+      "test_tanh_example",
+      "test_tanh",
       "test_tile",
       "test_tile_precomputed",
       "test_transpose_all_permutations_0",