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

Fix no bias

parent 45d12a67
No related branches found
No related tags found
2 merge requests!39Update 0.2.1 -> 0.3.0,!38Fix no bias
Pipeline #70708 passed
...@@ -30,7 +30,7 @@ void fullyconnected_forward ( ...@@ -30,7 +30,7 @@ void fullyconnected_forward (
// Solution: Add a system to check dataformat // Solution: Add a system to check dataformat
for (int och = 0; och < NB_OUTPUTS; och++) { for (int och = 0; och < NB_OUTPUTS; och++) {
Bias_T weightedSum = biases[och]; Bias_T weightedSum = (biases) ? biases[och] : Bias_T(0);
for (int iy = 0; iy < CHANNELS_HEIGHT; ++iy) { for (int iy = 0; iy < CHANNELS_HEIGHT; ++iy) {
for (int ix = 0; ix < CHANNELS_WIDTH; ++ix) { for (int ix = 0; ix < CHANNELS_WIDTH; ++ix) {
...@@ -48,7 +48,7 @@ Here the kernel to use with inputs in NHWC and weights in NHWC ...@@ -48,7 +48,7 @@ Here the kernel to use with inputs in NHWC and weights in NHWC
#pragma omp parallel for #pragma omp parallel for
for (int och = 0; och < NB_OUTPUTS; och++) { for (int och = 0; och < NB_OUTPUTS; och++) {
Bias_T weightedSum = biases[och]; Bias_T weightedSum = (biases) ? biases[och] : Bias_T(0);
for (int iy = 0; iy < CHANNELS_HEIGHT; ++iy) { for (int iy = 0; iy < CHANNELS_HEIGHT; ++iy) {
const int iPos = (CHANNELS_WIDTH * iy); const int iPos = (CHANNELS_WIDTH * iy);
......
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