Skip to content
Snippets Groups Projects
Commit ebd2a394 authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Fixed wrong offset

parent 681d419d
No related branches found
No related tags found
1 merge request!51Improve export
Pipeline #75474 failed
......@@ -30,7 +30,7 @@ void fullyconnected_forward (
Bias_T weightedSum = (biases) ? biases[och] : Bias_T(0);
for (int iy = 0; iy < CHANNELS_HEIGHT; ++iy) {
const int iOffset = (CHANNELS_WIDTH * iy);
const int iOffset = (NB_CHANNELS * CHANNELS_WIDTH * iy);
const int wOffset = NB_CHANNELS * CHANNELS_WIDTH
* (iy + CHANNELS_HEIGHT * och);
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment