Skip to content

Commit

Permalink
Merge pull request #1522 from zhouleiMax/develop
Browse files Browse the repository at this point in the history
fix conv2d_transpose op
  • Loading branch information
Jiang-Jia-Jun authored Feb 27, 2025
2 parents 7b0c467 + 128f45a commit 903a975
Showing 1 changed file with 8 additions and 2 deletions.
10 changes: 8 additions & 2 deletions paddle2onnx/mapper/nn/conv2d_transpose.h
Original file line number Diff line number Diff line change
Expand Up @@ -29,13 +29,19 @@ class Conv2dTransposeMapper : public Mapper {
GetAttr("dilations", &dilations_);
GetAttr("strides", &strides_);
GetAttr("paddings", &paddings_);
GetAttr("padding_algorithm", &padding_algorithm_);
if (HasAttr("padding_algorithm")) {
GetAttr("padding_algorithm", &padding_algorithm_);
} else {
padding_algorithm_ = "EXPLICIT";
}
GetAttr("data_format", &data_format_);

if (HasAttr("output_padding")) {
GetAttr("output_padding", &output_padding_);
}
GetAttr("output_size", &output_size_);
if (HasAttr("output_size")) {
GetAttr("output_size", &output_size_);
}
if (paddings_.size() == 2) {
paddings_.push_back(paddings_[0]);
paddings_.push_back(paddings_[1]);
Expand Down

0 comments on commit 903a975

Please sign in to comment.