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

fixed: maxpoolwithargmax's output1 have wrong shape, internal id: I7d5aeab58038bacb73373a4ff4f48a12bb6441db

This commit is contained in:
qin.chen 2022-07-27 14:21:59 +08:00 committed by Sven
parent 32241dc4ad
commit 84d76e5251
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/tim/vx/internal/src/ops/vsi_nn_op_maxpoolwithargmax.c
View File

@ -185,9 +185,12 @@ static vsi_bool op_setup
{
ret = vsi_nn_OpSetup( VSI_NN_OP_POOL, self, inputs, outputs );
outputs[1]->attr.dim_num = outputs[0]->attr.dim_num;
memcpy( outputs[1]->attr.size, outputs[0]->attr.size,
VSI_NN_MAX_DIM_NUM * sizeof(vsi_size_t) );
}
if ( VSI_NN_DIM_AUTO == outputs[1]->attr.dim_num )
{
outputs[1]->attr.dim_num = outputs[0]->attr.dim_num;
memcpy( outputs[1]->attr.size, outputs[0]->attr.size,
VSI_NN_MAX_DIM_NUM * sizeof(vsi_size_t) );
}
return ret;