@@ -13,7 +13,7 @@ int conv1d_lr(float *output_signal, unsigned out_time, unsigned out_channels, co
13
13
const ConvLayers_LR_Params * tparams = (ConvLayers_LR_Params * ) params ;
14
14
15
15
float * tempW = (float * )malloc (out_channels * in_channels * kernel_size * sizeof (float ));
16
- matmul (tempW , tparams -> W1 , tparams -> W2 , tparams -> rank , out_channels , in_channels * kernel_size );
16
+ matmul (tparams -> W1 , tparams -> W2 , out_channels , tparams -> rank , in_channels * kernel_size , 0 , 1.0 , tempW );
17
17
// Perform the Convolution
18
18
for (int t = 0 ; t < out_time ; t ++ ) {
19
19
for (int co = 0 ; co < out_channels ; co ++ ) {
@@ -48,7 +48,7 @@ int conv1d_depth_lr(float *output_signal, unsigned out_time, const float *input_
48
48
const ConvLayers_LR_Params * tparams = (ConvLayers_LR_Params * ) params ;
49
49
50
50
float * tempW = (float * )malloc (in_channels * kernel_size * sizeof (float ));
51
- matmul (tempW , tparams -> W1 , tparams -> W2 , tparams -> rank , in_channels , kernel_size );
51
+ matmul (tparams -> W1 , tparams -> W2 , in_channels , tparams -> rank , kernel_size , 0 , 1.0 , tempW );
52
52
// Perform the Convolution
53
53
for (int t = 0 ; t < out_time ; t ++ ) {
54
54
for (int ci = 0 ; ci < in_channels ; ci ++ ) {
0 commit comments