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

Merge branch 'batchnorm-fix' into 'dev'

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

See merge request !72
parents 566e26e0 f4e8fe10
No related branches found
No related tags found
Loading
Pipeline #82430 passed
......@@ -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