search

qjia7 / tfjs-core

WebGL-accelerated ML // linear algebra // automatic differentiation for JavaScript.
https://js.tensorflow.org
Apache License 2.0
2 stars 1 forks source link

tfjs-examples/mnist could not run correctly on compute-context #5

Open xhcao opened 5 years ago

xhcao commented 5 years ago

tfjs-examples/mnist could not run correctly on compute-context. It is related with that some operators could not work normally on compute-context. These operators are as bleow, batchMatMul, reduce, realDivide, add, subtract, exp, relu, conv2d, unpackTensor

GreyZzzzzzXh commented 5 years ago

Hi @xhcao , please revert this commit dced9250553aa6e26479a50a9c12e528d25ade4a and try again. This issue may be also caused by shared memory.

GreyZzzzzzXh commented 5 years ago

I found switching all ops used here to run on CS backend can solve this problem.