commit f22bddebff9238fd6d93722d3c05628cf692da64
author Alex Van Damme <atv@google.com> Mon Jun 17 16:27:49 2024 -0700
committer Alex Van Damme <atv@google.com> Mon Jun 17 16:38:39 2024 -0700
tree b05d02f8d00a8099f24b18cf6238b83e8dbbb2a1
parent 108cf326e6232b9afb2ad60ddf1b5e374e737319

Fix in_channel calculation in ConvS8D4

BUG=345095360

Change-Id: I9692c4d9c1c3c0e8e80b2084bcfc02ae1220e525

tflm/opt/conv_s8_d4.cc

diff --git a/tflm/opt/conv_s8_d4.cc b/tflm/opt/conv_s8_d4.cc
index 4ee4d33..9ee6edb 100644
--- a/tflm/opt/conv_s8_d4.cc
+++ b/tflm/opt/conv_s8_d4.cc
@@ -169,7 +169,7 @@
           acset_v(v48, v48);
           int in_channel = 0;
           do {
-            int in_channels_this_iter = std::min(filter_input_depth, 32);
+            int in_channels_this_iter = std::min(filter_input_depth - in_channel, 32);
             // Calculate first valid filter_y
             int filter_y = 0;
             {
@@ -456,7 +456,7 @@
 
           int in_channel = 0;
           while (in_channel < filter_input_depth) {
-            int in_channels_this_iter = std::min(filter_input_depth, 32);
+            int in_channels_this_iter = std::min(filter_input_depth - in_channel, 32);
             // Calculate first valid filter_y
             int filter_y = 0;
             {
@@ -556,7 +556,7 @@
           acset_v(v48, v48);
           int in_channel = 0;
           while (in_channel < filter_input_depth) {
-            int in_channels_this_iter = std::min(filter_input_depth, 32);
+            int in_channels_this_iter = std::min(filter_input_depth - in_channel, 32);
             cmds.conv.stop = (in_channels_this_iter / 4) - 1;
 
             // Calculate first valid filter_y
@@ -693,7 +693,7 @@
           int in_channel = 0;
 
           while (in_channel < filter_input_depth) {
-            int in_channels_this_iter = std::min(filter_input_depth, 32);
+            int in_channels_this_iter = std::min(filter_input_depth - in_channel, 32);
             // Calculate first valid filter_y
             int filter_y = 0;
             {