From 06b88e794078f77a3508ae70ca5e7ec7de498de0 Mon Sep 17 00:00:00 2001 From: Feiyue Chen Date: Wed, 14 Dec 2022 09:58:32 +0800 Subject: [PATCH] Fixed nn_params in groupconv2d Type: Bug Fix Signed-off-by: Feiyue Chen --- src/tim/vx/ops/groupedconv2d.cc | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/src/tim/vx/ops/groupedconv2d.cc b/src/tim/vx/ops/groupedconv2d.cc index f04f8a5..055c450 100644 --- a/src/tim/vx/ops/groupedconv2d.cc +++ b/src/tim/vx/ops/groupedconv2d.cc @@ -41,12 +41,12 @@ GroupedConv2d::GroupedConv2d(Graph* graph, padding_(padding), strides_(strides), dilation_(dilation), pad_({0,0,0,0}), group_number_(group_number), kernel_layout_(kernel_layout) { - this->impl()->node()->nn_param.conv2d.stride[0] = strides_[0]; - this->impl()->node()->nn_param.conv2d.stride[1] = strides_[1]; - this->impl()->node()->nn_param.conv2d.pad_type = TranslatePadType(padding_); - this->impl()->node()->nn_param.conv2d.group = group_number_; - this->impl()->node()->nn_param.conv2d.dilation[0] = dilation_[0]; - this->impl()->node()->nn_param.conv2d.dilation[1] = dilation_[1]; + this->impl()->node()->nn_param.grouped_conv2d.stride[0] = strides_[0]; + this->impl()->node()->nn_param.grouped_conv2d.stride[1] = strides_[1]; + this->impl()->node()->nn_param.grouped_conv2d.pad_type = TranslatePadType(padding_); + this->impl()->node()->nn_param.grouped_conv2d.group = group_number_; + this->impl()->node()->nn_param.grouped_conv2d.dilation[0] = dilation_[0]; + this->impl()->node()->nn_param.grouped_conv2d.dilation[1] = dilation_[1]; } GroupedConv2d::GroupedConv2d(Graph* graph,