IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[PT FE] Fix pad default value (#20401)

Browse Source
This commit is contained in:
Maxim Vafin
2023-10-12 13:28:03 +04:00
committed by GitHub
parent 2604c33d1b
commit 3455580780
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/frontends/pytorch/src/transforms/prim_list_construct_pad.cpp
+7 -1
View File
@@ -71,13 +71,19 @@ PrimListConstructPadReplacer::PrimListConstructPadReplacer() {
input_node = pad_op->input_value(0);
padding = pad_op->input_value(1);
auto mode_node = pad_op->input_value(2).get_node_shared_ptr();
pad_value = pad_op->input_value(3);
if (const auto& fw_node_mode = cast_fw_node(mode_node, "prim::Constant")) {
const auto& attrs = fw_node_mode->get_attrs();
if (attrs.find("string_value") != attrs.end()) {
mode = attrs.at("string_value");
}
}
pad_value = pad_op->input_value(3);
if (const auto& fw_node_mode = cast_fw_node(pad_value.get_node_shared_ptr(), "prim::Constant")) {
const auto& attrs = fw_node_mode->get_attrs();
if (attrs.find("none_value") != attrs.end()) {
pad_value = v0::Constant::create(element::f32, Shape{}, {0});
}
}
} else if ((pad_op = cast_fw_node(m.get_match_root(), "aten::reflection_pad2d"))) {
mode = "reflect";
input_node = pad_op->input_value(0);