提交 3ba2d55f 编写于 作者: C chenjiaoAngel

add openmp, test=develop

上级 c61cad38
...@@ -86,6 +86,7 @@ void pooling_basic(const float* din, ...@@ -86,6 +86,7 @@ void pooling_basic(const float* din,
} }
} else { } else {
for (int ind_n = 0; ind_n < num; ++ind_n) { for (int ind_n = 0; ind_n < num; ++ind_n) {
#pragma omp parallel for
for (int ind_c = 0; ind_c < chin; ++ind_c) { for (int ind_c = 0; ind_c < chin; ++ind_c) {
for (int ind_h = 0; ind_h < hout; ++ind_h) { for (int ind_h = 0; ind_h < hout; ++ind_h) {
int sh = ind_h * stride_h; int sh = ind_h * stride_h;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册