| | |
|---|
| | | mean_cpu(l.output, l.batch, l.out_c, l.out_h*l.out_w, l.mean); |
|---|
| | | variance_cpu(l.output, l.mean, l.batch, l.out_c, l.out_h*l.out_w, l.variance); |
|---|
| | | |
|---|
| | | scal_cpu(l.out_c, .99, l.rolling_mean, 1); |
|---|
| | | axpy_cpu(l.out_c, .01, l.mean, 1, l.rolling_mean, 1); |
|---|
| | | scal_cpu(l.out_c, .99, l.rolling_variance, 1); |
|---|
| | | axpy_cpu(l.out_c, .01, l.variance, 1, l.rolling_variance, 1); |
|---|
| | | scal_cpu(l.out_c, .9, l.rolling_mean, 1); |
|---|
| | | axpy_cpu(l.out_c, .1, l.mean, 1, l.rolling_mean, 1); |
|---|
| | | scal_cpu(l.out_c, .9, l.rolling_variance, 1); |
|---|
| | | axpy_cpu(l.out_c, .1, l.variance, 1, l.rolling_variance, 1); |
|---|
| | | |
|---|
| | | copy_cpu(l.outputs*l.batch, l.output, 1, l.x, 1); |
|---|
| | | normalize_cpu(l.output, l.mean, l.variance, l.batch, l.out_c, l.out_h*l.out_w); |
|---|
