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

Fix groupconv2d pad parameter

This commit is contained in:
Your Name 2021-07-29 15:07:22 +08:00 committed by Sven
parent a09ffe8b98
commit 70c427256d
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -63,6 +63,10 @@ GroupedConv2d::GroupedConv2d(Graph* graph,
this->impl()->node()->nn_param.conv2d.group = group_number_; 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[0] = dilation_[0];
this->impl()->node()->nn_param.conv2d.dilation[1] = dilation_[1]; this->impl()->node()->nn_param.conv2d.dilation[1] = dilation_[1];
this->impl()->node()->nn_param.conv2d.pad[0] = pad_[0];
this->impl()->node()->nn_param.conv2d.pad[1] = pad_[1];
this->impl()->node()->nn_param.conv2d.pad[2] = pad_[2];
this->impl()->node()->nn_param.conv2d.pad[3] = pad_[3];
} }
std::shared_ptr<Operation> GroupedConv2d::Clone( std::shared_ptr<Operation> GroupedConv2d::Clone(