提交 4207db16 authored 作者: Frederic Bastien's avatar Frederic Bastien

small optimization in the cpu conv up.

上级 d0184177
...@@ -1606,7 +1606,7 @@ for(int b=0;b< %(self_bsize)s;b++){ ...@@ -1606,7 +1606,7 @@ for(int b=0;b< %(self_bsize)s;b++){
const %(type)s * idx_in=&in[ind0*dim_im[1]]; const %(type)s * idx_in=&in[ind0*dim_im[1]];
if(iter_n + 4*%(self_dy)s < dim_zz[1] if(iter_n + 4*%(self_dy)s < dim_zz[1]
&& iter_n>dim_ker[1]-1+3 && iter_n>dim_ker[1]-1
&& iter_n<dim_im[1]-dim_ker[1]+1-3){ && iter_n<dim_im[1]-dim_ker[1]+1-3){
nb_sum=4; nb_sum=4;
for (int ind1=pos_n-k; k<max_k; k++,ind1--) { for (int ind1=pos_n-k; k<max_k; k++,ind1--) {
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论