diff --git a/src/operator/image/image_random-inl.h b/src/operator/image/image_random-inl.h index 64f12b4f3a6c..5041bf433588 100644 --- a/src/operator/image/image_random-inl.h +++ b/src/operator/image/image_random-inl.h @@ -163,11 +163,11 @@ template struct normalize_forward { template MSHADOW_XINLINE static void Map(uint32_t c, DType* out_data, const DType* in_data, - const NormalizeParam ¶m, const int length, + const NormalizeParam ¶m, const int length, const int step) { DType mean = param.mean[param.mean.ndim() > c ? c : 0]; DType std_dev = param.std[param.std.ndim() > c ? c : 0]; - + #pragma omp parallel for for (int i = 0; i < length; ++i) { KERNEL_ASSIGN(out_data[step + c*length + i], req,