In the page you write “Due to a limitation with Tensorflow.js in the browser, model prediction can only happen on the CPU.” But all ops in TensorFlow JS should be using WebGL, and so why are the predictions only happening on the CPU in your case?
In the page you write “Due to a limitation with Tensorflow.js in the browser, model prediction can only happen on the CPU.”
But all ops in TensorFlow JS should be using WebGL, and so why are the predictions only happening on the CPU in your case?