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
102 stars 18 forks source link

Check output shape #161

Closed BruceDai closed 2 years ago

BruceDai commented 2 years ago

@huningxin @mingmingtasd @miaobin PTAL, thanks.

miaobin commented 2 years ago

Great work! Thanks ! @BruceDai I reviewed for most of ops except the Conv2dTranspose and Gru. Please help review again and especially focus on these two. @miaobin

Great work! I've meet some issues with formula comprehension and have discussed with @BruceDai LGTM, Thanks!

BruceDai commented 2 years ago

Thanks @mingmingtasd and @miaobin for review this patch, let's merge it.