From 18749f5d05c106381082e76fef36f2894578ca98 Mon Sep 17 00:00:00 2001 From: Chen Feiyue <69809761+chenfeiyue-cfy@users.noreply.github.com> Date: Sat, 8 Jul 2023 23:40:32 +0800 Subject: [PATCH] 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 --- src/tim/vx/internal/src/ops/vsi_nn_op_deconvolution1d.c | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/tim/vx/internal/src/ops/vsi_nn_op_deconvolution1d.c b/src/tim/vx/internal/src/ops/vsi_nn_op_deconvolution1d.c index 1180dbe..494d3d4 100644 --- a/src/tim/vx/internal/src/ops/vsi_nn_op_deconvolution1d.c +++ b/src/tim/vx/internal/src/ops/vsi_nn_op_deconvolution1d.c @@ -264,9 +264,8 @@ static vsi_bool op_setup } 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.dim_num = inputs[0]->attr.dim_num; }