Skip to content

Commit

Permalink
Fixing specialization of DenseT<T, 1, N> for XSIMD backend
Browse files Browse the repository at this point in the history
  • Loading branch information
jatinchowdhury18 committed Feb 21, 2024
1 parent e14dc6e commit 89b8c8b
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion RTNeural/dense/dense_xsimd.h
Original file line number Diff line number Diff line change
Expand Up @@ -338,8 +338,9 @@ class DenseT<T, 1, out_sizet>
for(int i = 0; i < v_out_size; ++i)
outs[i] = bias[i];

const auto in = ins[0].get (0);
for(int i = 0; i < v_out_size; ++i)
outs[i] += ins[0] * weights[i];
outs[i] += in * weights[i];
}

/**
Expand Down

0 comments on commit 89b8c8b

Please sign in to comment.