[webgpu] Fix wrongly used u32 of WGSL#5559

lina128 merged 2 commits intotensorflow:mastertensorflow/tfjs:masterfrom

axinging:fix_u32_wgslaxinging/tfjs:fix_u32_wgslCopy head branch name to clipboard