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(