[WebGL] Failed to compile fragment shader

System information

• Have I written custom code (as opposed to using a stock example script provided in TensorFlow.js): Yes
• OS Platform and Distribution (e.g., Linux Ubuntu 16.04): All platforms
• TensorFlow.js version (use command below): tf.js 3.7.0 webgl backend.
• Browser version: Chrome 93.0.4545.0

Describe the current behavior WebGL backend throws Failed to compile fragment shader error while using depthwiseConv2d op with specific options. While WebGPU backend doesn’t reproduce.

Describe the expected behavior No error.

Standalone code to reproduce the issue

CodePen: https://codepen.io/honry/pen/wvdMmva

    const input = tf.fill([1, 65, 65, 960], 0);
    const weights = tf.fill([3, 3, 960, 1], 1);
    const result = tf.depthwiseConv2d(input, weights, 1, 'same', 'NHWC', 4);

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.

Please check error log from https://codepen.io/honry/pen/wvdMmva

[Honry]

@huningxin, this issue only reproduce on webgl backend.

[Honry]

Thanks @rthadur, I’ve created a new issue at https://github.com/tensorflow/tfjs/issues/5378.