From 84d76e5251e3e616ac1141064ce8794c4c49cce2 Mon Sep 17 00:00:00 2001 From: "qin.chen" Date: Wed, 27 Jul 2022 14:21:59 +0800 Subject: [PATCH] fixed: maxpoolwithargmax's output1 have wrong shape, internal id: I7d5aeab58038bacb73373a4ff4f48a12bb6441db --- .../vx/internal/src/ops/vsi_nn_op_maxpoolwithargmax.c | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/tim/vx/internal/src/ops/vsi_nn_op_maxpoolwithargmax.c b/src/tim/vx/internal/src/ops/vsi_nn_op_maxpoolwithargmax.c index 3432790..23b987d 100644 --- a/src/tim/vx/internal/src/ops/vsi_nn_op_maxpoolwithargmax.c +++ b/src/tim/vx/internal/src/ops/vsi_nn_op_maxpoolwithargmax.c @@ -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;