Skip to content

Commit 66c3b41

Browse files
Rectified the averaging operation
1 parent 590be76 commit 66c3b41

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

c_reference/src/Makefile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ include ../config.mk
66
INCLUDE_DIR=../include
77
IFLAGS = -I $(INCLUDE_DIR)
88

9-
all: utils.o fastgrnn.o classifier.o rnnpool.o quantized_utils.o quantized_fastgrnn.o quantized_rnnpool.o quantized_mbconv.o conv1d.o conv_utils.o
9+
all: conv1d.o conv_utils.o utils.o fastgrnn.o classifier.o rnnpool.o quantized_utils.o quantized_fastgrnn.o quantized_rnnpool.o quantized_mbconv.o
1010

1111
conv_utils.o : conv_utils.c
1212
$(CC) -o $@ $(IFLAGS) $(CFLAGS) -c $^

c_reference/src/conv1d.c

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -193,16 +193,16 @@ int AvgPool1D(float *output_signal, unsigned out_T, const float *input_signal, u
193193
sum += (input_signal[((tf + t) - padding) * in_channels + ci]);
194194
}
195195
if(activations == 1){
196-
output_signal[t * in_channels + ci] = sigmoid(sum);
196+
output_signal[t * in_channels + ci] = sigmoid(sum/(float)kernel_size);
197197
}
198198
else if(activations == 2){
199-
output_signal[t * in_channels + ci] = tanh(sum);
199+
output_signal[t * in_channels + ci] = tanh(sum/(float)kernel_size);
200200
}
201201
else if(activations == 3){
202-
output_signal[t * in_channels + ci] = relu(sum);
202+
output_signal[t * in_channels + ci] = relu(sum/(float)kernel_size);
203203
}
204204
else{
205-
output_signal[t * in_channels + ci] = sum;
205+
output_signal[t * in_channels + ci] = sum/(float)kernel_size;
206206
}
207207

208208
}

0 commit comments

Comments
 (0)