Skip to content
Snippets Groups Projects
Commit f4e8fe10 authored by Clément Fisher's avatar Clément Fisher
Browse files

Modification of the forward method of the batchnorm.hpp file to match the...

Modification of the forward method of the batchnorm.hpp file to match the order of Producers defined in aidge_core.
parent 566e26e0
No related branches found
No related tags found
1 merge request!72Modification of the forward method of the batchnorm.hpp file to match the...
Pipeline #83450 failed
......@@ -18,10 +18,10 @@ __attribute__((always_inline)) inline
void batchnorm_forward (
const Input_T* __restrict inputs,
Output_T* __restrict outputs,
const Param_T* __restrict scales,
const Param_T* __restrict biases,
const Param_T* __restrict variances,
const Param_T* __restrict means,
const Param_T* __restrict scales,
const Param_T* __restrict variances,
const double epsilon,
const Rescaling_T& __restrict rescaling)
{
......
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