13:I[57967,["/_next/static/chunks/295d483fd0fe7058.js","/_next/static/chunks/09e0c938617bc9e1.js","/_next/static/chunks/680edad3696a354b.js"],"Project"] 14:T4e4,const conv_width:i32= 3; fn u(s:texture_2d, coord:vec2i, fc:vec2u)->vec4f{ return textureLoad(s, vec2i(fc)+ coord+ vec2i(1),0);// padding } fn conv(s:texture_2d,K:mat3x3, fc:vec2u)->vec4f{ var x=vec4f(0); for (var i = -conv_width/2; i < conv_width/2+1; i=i+1){ for (var j = -conv_width/2; j < conv_width/2+1; j=j+1){ x += K[i+conv_width/2][j+conv_width/2] * u(s,vec2i(i,j),fc); } } return x; } fn activation(x:f32)->f32{ return max(0,x*0.5); } fn dactivation(x:f32)->f32{ return 0.5*step(0,x); } fn sigmoid(x:vec4f)->vec4f{ return 1/(1+exp(-x*.1)); } fn getK(s:texture_2d_array,dim:u32)->mat3x3{ var K=mat3x3(); for(var i:u32=0;i<3;i++){ for(var j:u32=0;j<3;j++){ K[i][j]=textureLoad(s,vec2u(i,j),dim,0).r; } } return K; } @compute @workgroup_size(1) fn main(@builtin(global_invocation_id) id : vec3u){ var K = getK(iChannel2,id.z); var b = textureLoad(iChannel2,vec2u(0,0),id.z,0).a; var C = conv(iChannel1,K,id.xy);//+b; C=vec4f(activation(C.x));//max(C*.5,vec4f(0)); //activation //C=sigmoid(C+b); //activation // Write the color to the texture textureStore(oChannel, id.xy, id.z,vec4f(C.xyz,1)); }15:T47b,const conv_width:i32= 3; fn uA(s:texture_2d_array, coord:vec2i, fc:vec2u, dim:u32)->vec4f{ return textureLoad(s, vec2i(fc)+ coord+ vec2i(1),dim,0);// padding } fn convA(s:texture_2d_array,K:mat3x3, fc:vec2u, dim:u32)->vec4f{ var x=vec4f(0); for (var i = -conv_width/2; i < conv_width/2+1; i=i+1){ for (var j = -conv_width/2; j < conv_width/2+1; j=j+1){ x += K[i+conv_width/2][j+conv_width/2] * uA(s,vec2i(i,j),fc,dim); } } return x; } fn getK(s:texture_2d_array,dim:u32)->mat3x3{ var K=mat3x3(); for(var i:u32=0;i<3;i++){ for(var j:u32=0;j<3;j++){ K[i][j]=textureLoad(s,vec2u(i,j),dim,0).r; } } return K; } @compute @workgroup_size(1) fn main(@builtin(global_invocation_id) id : vec3u){ var K=mat3x3(); for(var i:u32=0;i<32;i++){ K += getK(iChannel2,id.z); } var C=convA(iChannel1,K,id.xy,id.z); C=max(C*.5,vec4f(0)); //activation //C=max(C,vec4f(0)); //activation //TODO handle bias in alpha channel // Write the color to the texture textureStore(oChannel, id.xy, id.z,vec4f(C.xyz,1)); }16:Ta13,