WebGL CPU forwarding failed for slice op

Please make sure that this is a bug. As per our GitHub Policy, we only address code/doc bugs, performance issues, feature requests and build/installation issues on GitHub. tag:bug_template

System information

• Have I written custom code (as opposed to using a stock example script provided in TensorFlow.js):
• OS Platform and Distribution (e.g., Linux Ubuntu 16.04):
• Mobile device (e.g. iPhone 8, Pixel 2, Samsung Galaxy) if the issue happens on mobile device:
• TensorFlow.js installed from (npm or script link): both
• TensorFlow.js version (use command below): 2.5.0
• Browser version:
• Tensorflow.js Converter Version: 2.5.0

Describe the current behavior WebGL CPU forward failed to slice op, caused by CPU slice op has been modularized.

backend.ts:665 Uncaught (in promise) Error: 'slice' not yet implemented or not found in the registry. This kernel may not be supported by the tfjs backend you have chosen
    at notYetImplemented (backend.ts:665)
    at MathBackendCPU.KernelBackend.slice (backend.ts:145)
    at MathBackendWebGL.slice (backend_webgl.ts:731)
    at forward (slice.ts:75)
    at engine.ts:625

The current work-around is to disable CPU forwarding:

tf.env().set('WEBGPU_CPU_FORWARD', false);

Describe the expected behavior In 2.4.0 the CPU forwarding of slice op is not failing.

Standalone code to reproduce the issue Provide a reproducible test case that is the bare minimum necessary to generate the problem. If possible, please share a link to Colab/CodePen/any notebook.

Other info / logs Include any logs or source code that would be helpful to diagnose the problem. If including tracebacks, please include the full traceback. Large logs and files should be attached.

[lina128]

We published 2.6.0 today, it should fix the problem by now.

[mtuan4i]

I got the same error. I tried tf.env().set('WEBGPU_CPU_FORWARD', false); but it doesn’t work. However, tf.env().set('WEBGL_CPU_FORWARD', false); does