Adding new features

2024-07-23 13:00:49 +03:00
parent 9e044fd7fc
commit 745cc4ed6d
28 changed files with 33632 additions and 106 deletions
@@ -0,0 +1,70 @@
+#ifndef DWS_CONV_H
+#define DWS_CONV_H
+
+void pw_conv(int in_dim[3], int fil_dim[4], int out_dim[3], int inp[in_dim[0]][in_dim[1]][in_dim[2]], 
+	      const int fil[fil_dim[0]][fil_dim[3]], const int bias[], 
+	      int out[out_dim[0]][out_dim[1]][out_dim[2]], int strides, int pad[], 
+              const int bias_shift_mode, const int quantized_multiplier, const int out_shift_rl){
+
+     int i, j, k, m, res, str1, str2, quant_prod;
+
+     for (i = 0; i < out_dim[2]; i++) {   // output depth
+           str1 = -pad[0] - strides;
+           for (j = 0; j < out_dim[0]; j++) {  // output height
+	        str1 += strides;
+	        str2 = -pad[2] - strides;
+	        for (k = 0; k < out_dim[1]; k++) {  // output width
+                    res = bias[i];
+                      str2 += strides;
+                      if (str1 < in_dim[0] && str1 >= 0 && str2 >= 0 && str2 < in_dim[1]) { 
+		           for (m = 0; m < fil_dim[3]; m++) {   // filters depth
+                                res += inp[str1][str2][m] * fil[i][m];
+                          }
+                      }
+                      quant_prod = quantized_multiplier * res + (1 << (out_shift_rl -1));
+        	      quant_prod = quant_prod >> (out_shift_rl);
+        	      if(quant_prod < 0) quant_prod = 0;
+        	      if(quant_prod > 255) quant_prod = 255;
+                      out[j][k][i] = quant_prod;
+	       }
+          }
+     }
+}
+
+void dw_conv(int in_dim[3], int depthwise_fil_dim[4], int out_dim[3], int inp[in_dim[0]][in_dim[1]][in_dim[2]], 
+            const int depthwise_fil[depthwise_fil_dim[0]][depthwise_fil_dim[1]][depthwise_fil_dim[2]][1], const int bias[], 
+			int out[out_dim[0]][out_dim[1]][out_dim[2]], int strides, int pad[], 
+            const int bias_shift_mode, const int depthwise_multiplier, const int depthwise_out_shift_rl){
+    
+	int i, j, k, n, p, res, k1, k2, str1, str2, quant_prod;
+
+	// Depthwise convolution
+	for (i = 0; i < out_dim[2]; i++) {   // output depth
+		str1 = -pad[0] - strides;
+		for (j = 0; j < out_dim[0]; j++) {  // output height
+			str1 += strides;
+			str2 = -pad[2] - strides;
+			for (k = 0; k < out_dim[1]; k++) {  // output width
+				res = bias[i];
+				str2 += strides;
+				for (p = 0; p < depthwise_fil_dim[1]; p++){  // depthwise filter height
+					for (n = 0; n < depthwise_fil_dim[2]; n++) {  // depthwise filter width
+						k1 = str1 + p; 
+						k2 = str2 + n;
+                        
+						if (k1 < in_dim[0] && k1 >= 0 && k2 >= 0 && k2 < in_dim[1]) { 
+							res += inp[k1][k2][i] * depthwise_fil[i][p][n][0];
+						}
+					}
+				}
+				quant_prod = depthwise_multiplier * res + (1 << (depthwise_out_shift_rl -1));
+		        quant_prod = quant_prod >> (depthwise_out_shift_rl);
+				if(quant_prod < 0) quant_prod = 0;
+        		if(quant_prod > 255) quant_prod = 255;
+                out[j][k][i] = quant_prod;
+            }
+		}
+	}
+}
+
+#endif  /* DWS_CONV_H */