search

webmachinelearning / webnn-polyfill

🧠⚙️ Web Neural Network API polyfill based on TensorFlow.js
https://www.npmjs.com/package/@webmachinelearning/webnn-polyfill
Apache License 2.0
101 stars 18 forks source link

Style Transfer failed with WebGL backend #174

Closed Honry closed 2 years ago

Honry commented 2 years ago

Sample: https://webmachinelearning.github.io/webnn-samples/style_transfer/index.html

Wasm backend works, while WebGL backend throws error: Binary was compiled with different shapes than the current args. Shapes 1,8,270,270 and 1,270,270,8 must match from convTranspose2d op compilation.

BruceDai commented 2 years ago

Thanks @Honry, I will investigate and fix this issue.