未验证 提交 865728c6 编写于 作者: E eclipsycn 提交者: GitHub

Update batchnorm_arm_func.h

上级 8b57d149
...@@ -53,7 +53,7 @@ void BatchnormCompute(const BatchNormParam &param) { ...@@ -53,7 +53,7 @@ void BatchnormCompute(const BatchNormParam &param) {
"C must equal to variance.numel()"); "C must equal to variance.numel()");
int HXW = H * W; int HXW = H * W;
if (0 && HXW > 32) { if (HXW > 32) {
int NXC = N * C; int NXC = N * C;
float *inv_std_ptr = new float[NXC * 4]; float *inv_std_ptr = new float[NXC * 4];
float *volatile new_scale_ptr = new float[NXC * 4]; float *volatile new_scale_ptr = new float[NXC * 4];
...@@ -224,12 +224,6 @@ void BatchnormCompute(const BatchNormParam &param) { ...@@ -224,12 +224,6 @@ void BatchnormCompute(const BatchNormParam &param) {
} }
} }
} }
// for(int i = 0; i < new_scale.numel(); i++){
// std::cout << "new_scale " << new_scale_ptr[i] <<std::endl;
// }
// for(int i = 0; i < new_bias.numel(); i++){
// std::cout << "new_bias " << new_bias_ptr[i] <<std::endl;
// }
delete[] inv_std_ptr; delete[] inv_std_ptr;
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册