Fix potential shader compile issue in LeakyRelu op (#106)

microsoft · Mar 15, 2019 · 988f891 · 988f891
1 parent 3439e56
commit 988f891
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/backends/webgl/ops/leaky-relu.ts b/lib/backends/webgl/ops/leaky-relu.ts
@@ -19,7 +19,7 @@ export class WebGLLeakyRelu extends LeakyRelu implements WebGLOperator {
       uniform sampler2D A;
       void main() {
         float v = texture2D(A, TexCoords).r;
-        gl_FragColor = vec4(v < 0.0 ? v * ${this.alpha} : v);
+        gl_FragColor = vec4(v < 0.0 ? v * float(${this.alpha}) : v);
       }
       `;
     return {