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

Fixed nn_params in groupconv2d 

Type: Bug Fix
Signed-off-by: Feiyue Chen <Feiyue.Chen@verisilicon.com>
This commit is contained in:
Feiyue Chen 2022-12-14 09:58:32 +08:00 committed by Sven
parent c6919248e1
commit 06b88e7940
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/tim/vx/ops/groupedconv2d.cc
View File

@ -41,12 +41,12 @@ GroupedConv2d::GroupedConv2d(Graph* graph,
padding_(padding), strides_(strides), dilation_(dilation), padding_(padding), strides_(strides), dilation_(dilation),
pad_({0,0,0,0}), group_number_(group_number), pad_({0,0,0,0}), group_number_(group_number),
kernel_layout_(kernel_layout) { kernel_layout_(kernel_layout) {
this->impl()->node()->nn_param.conv2d.stride[0] = strides_[0]; this->impl()->node()->nn_param.grouped_conv2d.stride[0] = strides_[0];
this->impl()->node()->nn_param.conv2d.stride[1] = strides_[1]; this->impl()->node()->nn_param.grouped_conv2d.stride[1] = strides_[1];
this->impl()->node()->nn_param.conv2d.pad_type = TranslatePadType(padding_); this->impl()->node()->nn_param.grouped_conv2d.pad_type = TranslatePadType(padding_);
this->impl()->node()->nn_param.conv2d.group = group_number_; this->impl()->node()->nn_param.grouped_conv2d.group = group_number_;
this->impl()->node()->nn_param.conv2d.dilation[0] = dilation_[0]; this->impl()->node()->nn_param.grouped_conv2d.dilation[0] = dilation_[0];
this->impl()->node()->nn_param.conv2d.dilation[1] = dilation_[1]; this->impl()->node()->nn_param.grouped_conv2d.dilation[1] = dilation_[1];
} }
GroupedConv2d::GroupedConv2d(Graph* graph, GroupedConv2d::GroupedConv2d(Graph* graph,