Refactor pool3d function to simplify output handling

openvinotoolkit · Dec 24, 2023 · da3f73b · da3f73b
1 parent 9b57135
commit da3f73b
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/src/frontends/paddle/src/op/pool3d.cpp b/src/frontends/paddle/src/op/pool3d.cpp
@@ -207,13 +207,7 @@ NamedOutputs pool3d(const NodeContext& node) {
             std::make_shared<default_opset::Constant>(ov::element::i64, Shape{5}, std::vector<int64_t>{0, 2, 3, 4, 1}));
     }
 
-    auto output_name = node.get_output_names();
-    if (output_name.size() == 1) {
-        return NamedOutputs{{"Out", {pool_outputs[0]}}};
-    } else {
-        PADDLE_OP_CHECK(node, pool_outputs.size() == 2, "returnmask can`t use with global_pooling");
-        return NamedOutputs{{"Out", {pool_outputs[0]}}, {"Mask", {pool_outputs[1]}}};
-    }
+    return NamedOutputs{{"Out", {pool_outputs[0]}}};
 }
 
 NamedOutputs pool3d_with_index(const NodeContext& node) {