From 70c427256d732997fc45788f01772f8d1e4397a0 Mon Sep 17 00:00:00 2001 From: Your Name Date: Thu, 29 Jul 2021 15:07:22 +0800 Subject: [PATCH] Fix groupconv2d pad parameter --- src/tim/vx/ops/groupedconv2d.cc | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/tim/vx/ops/groupedconv2d.cc b/src/tim/vx/ops/groupedconv2d.cc index f66d0a4..040eaf6 100644 --- a/src/tim/vx/ops/groupedconv2d.cc +++ b/src/tim/vx/ops/groupedconv2d.cc @@ -63,6 +63,10 @@ GroupedConv2d::GroupedConv2d(Graph* graph, 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.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 GroupedConv2d::Clone(