update cnn code

1 files changed, 2 insertions, 2 deletions

diff --git a/workspaces/main/shaders/cnn/cnn_conv1x1.wgsl b/workspaces/main/shaders/cnn/cnn_conv1x1.wgsl
index d468182..f77cfa8 100644
--- a/workspaces/main/shaders/cnn/cnn_conv1x1.wgsl
+++ b/workspaces/main/shaders/cnn/cnn_conv1x1.wgsl
@@ -44,7 +44,7 @@ fn cnn_conv1x1_7to4_src(
 ) -> vec4<f32> {
   let step = 1.0 / resolution;
 
-  let original = (textureSample(tex, samp, uv) - 0.5) * 2.0;
+  var original = (textureSample(tex, samp, uv) - 0.5) * 2.0;
   let gray = dot(original.rgb, vec3<f32>(0.2126, 0.7152, 0.0722));
   let uv_norm = (uv - 0.5) * 2.0;
   let in1 = vec4<f32>(uv_norm, gray, 1.0);
@@ -55,7 +55,7 @@ fn cnn_conv1x1_7to4_src(
   for (var dy = -0; dy <= 0; dy++) {
     for (var dx = -0; dx <= 0; dx++) {
       let offset = vec2<f32>(f32(dx), f32(dy)) * step;
-      let rgbd = (textureSample(tex, samp, uv + offset) - 0.5) * 2.0;
+      var rgbd = (textureSample(tex, samp, uv + offset) - 0.5) * 2.0;
 
       sum.r += dot(weights[pos+0], rgbd) + dot(weights[pos+1], in1);
       sum.g += dot(weights[pos+2], rgbd) + dot(weights[pos+3], in1);