How to use data kept on GPU after model inference with tensorflow.js v3.13.0?

Hi! I’ve just been aware of the new feature to keep data on GPU with tfjs v3.13.0 made by @lina128 and I am very excited to try it!

However, I’m a bit confused on how to use it, since I cannot use the texture directly in my shader. From what I understand, I need to provide the canvas on which I will run my shader to the declaration of the WebGL backend, but I currently don’t declare any backend… Instead, I use the setBackend(string) method from @tensorflow/tfjs-core.

How should one declare the WebGL backend to which provide they would provide the canvas?

I’ve also created a question on StackOverflow, if that is the preferred way of proceeding. https://stackoverflow.com/questions/70703710/how-to-use-data-kept-on-gpu-after-model-inference-with-tensorflow-js-v3-13-0

Thanks

[lina128]

Hi, we will provide an example in tfjs-example shortly. Likely next week.

[lina128]

Glad it works. Good luck with your project, @RaphaelRoyerRivard !