colin
/
TIM-VX
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixed transient deconv1d generate wrong output shape bug (#619) 

Fixed bug that when deconv1d ouput is set to be transient, actual output shape will be zero at dim 1.

Reason :internal typing error
Type: Bug Fix
Signed-off-by: Feiyue Chen <Feiyue.Chen@verisilicon.com>
This commit is contained in:
Chen Feiyue 2023-07-08 23:40:32 +08:00 committed by GitHub
parent ea8046ec9c
commit 18749f5d05
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/tim/vx/internal/src/ops/vsi_nn_op_deconvolution1d.c
View File

@ -264,9 +264,8 @@ static vsi_bool op_setup
} }
else else
{ {
outputs[0]->attr.size[1] = inputs[1]->attr.size[3]; outputs[0]->attr.size[1] = inputs[1]->attr.size[2];
} }
outputs[0]->attr.size[1] = nn_param->weights;
outputs[0]->attr.size[2] = inputs[0]->attr.size[2]; outputs[0]->attr.size[2] = inputs[0]->attr.size[2];
outputs[0]->attr.dim_num = inputs[0]->attr.dim_num; outputs[0]->attr.dim_num = inputs[0]->attr.dim_num;
} }