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(data, element::i32)); value = context.mark_node(std::make_shared(value, data)); - auto broadcasted_value = context.mark_node(std::make_shared(value, data_shape)); auto bool_mask = context.mark_node(std::make_shared(mask, element::boolean)); - return {context.mark_node(std::make_shared(bool_mask, broadcasted_value, data))}; + return {context.mark_node(std::make_shared(bool_mask, value, data))}; }; } // namespace op } // namespace pytorch } // namespace frontend -} // namespace ov \ No newline at end of file +} // namespace ov