Skip to content
Snippets Groups Projects
Commit 435b6c51 authored by Houssem ROUIS's avatar Houssem ROUIS
Browse files

test3 reload CI

parent 3ce6135f
No related branches found
No related tags found
3 merge requests!15version 0.2.0,!12Lenetop,!10Lenet operators
This commit is part of merge request !10. Comments created here will be created in the context of that merge request.
...@@ -25,7 +25,7 @@ void fc_forward_cuda_kernel(std::size_t nbInputs, std::size_t inChannels, std::s ...@@ -25,7 +25,7 @@ void fc_forward_cuda_kernel(std::size_t nbInputs, std::size_t inChannels, std::s
for (std::size_t in = 0; in < inChannels; ++in) { for (std::size_t in = 0; in < inChannels; ++in) {
sum += input[batch * inChannels + in] * weights[out * inChannels + in]; sum += input[batch * inChannels + in] * weights[out * inChannels + in];
} }
output[batch * outChannels + out] = sum + (noBias ? 0 : bias[out]); output[batch * outChannels + out] = sum + (noBias ? 0 : bias[out]);
} }
} }
} }
......
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