From 1b5f4287524b2c33917baa9d5bebd2b3694b24d7 Mon Sep 17 00:00:00 2001
From: Xiuchuan Zhai <xiuchuan.zhai@intel.com>
Date: Tue, 5 Sep 2023 14:36:30 +0800
Subject: [PATCH] eliminate broadcast node in masked_fill (#19595)

---
 src/frontends/pytorch/src/op/masked_fill.cpp | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/src/frontends/pytorch/src/op/masked_fill.cpp b/src/frontends/pytorch/src/op/masked_fill.cpp
index 5ed090e0b61..2093e056790 100644
--- a/src/frontends/pytorch/src/op/masked_fill.cpp
+++ b/src/frontends/pytorch/src/op/masked_fill.cpp
@@ -23,14 +23,12 @@ OutputVector translate_masked_fill(const NodeContext& context) {
     auto data = context.get_input(0);
     auto mask = context.get_input(1);
     auto value = context.get_input(2);
-    auto data_shape = context.mark_node(std::make_shared<v3::ShapeOf>(data, element::i32));
     value = context.mark_node(std::make_shared<v1::ConvertLike>(value, data));
-    auto broadcasted_value = context.mark_node(std::make_shared<v3::Broadcast>(value, data_shape));
     auto bool_mask = context.mark_node(std::make_shared<v0::Convert>(mask, element::boolean));
-    return {context.mark_node(std::make_shared<v1::Select>(bool_mask, broadcasted_value, data))};
+    return {context.mark_node(std::make_shared<v1::Select>(bool_mask, value, data))};
 };
 
 }  // namespace op
 }  // namespace pytorch
 }  // namespace frontend
-}  // namespace ov
\ No newline at end of file
+}  // namespace ov