Skip to content

Commit ca28078

Browse files
Rectified the activations in TanhGate
1 parent a6d8836 commit ca28078

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

c_reference/src/conv1d.c

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -204,7 +204,6 @@ int AvgPool1D(float *output_signal, unsigned out_T, const float *input_signal, u
204204
else{
205205
output_signal[t * in_channels + ci] = sum/(float)kernel_size;
206206
}
207-
208207
}
209208
}
210209
return 0;

c_reference/src/conv_utils.c

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,10 +15,10 @@ int prepareLowRankConvMat(float* out, float* W1, float* W2, unsigned rank, unsig
1515
}
1616

1717
int TanhGate(float* output_signal, float* input_signal, unsigned in_T, unsigned in_channels){
18-
unsigned int piv = in_channels>>2;
18+
unsigned int piv = in_channels>>1;
1919
for(int t = 0 ; t < in_T ; t++){
2020
for(int d = 0 ; d < piv ; d++){
21-
output_signal[t * piv + d] = input_signal[t * in_channels + d] * input_signal[t * in_channels + (d + piv)];
21+
output_signal[t * piv + d] = sigmoid(input_signal[t * in_channels + d]) * tanh(input_signal[t * in_channels + (d + piv)]);
2222
}
2323
}
2424
return 0;

0 commit comments

Comments
 (0)