From ec8765f9504a365d16e0b62caf4da5200f6b12a5 Mon Sep 17 00:00:00 2001
From: Mateusz Mikolajczyk <mateusz.mikolajczyk@intel.com>
Date: Mon, 18 Dec 2023 14:31:56 +0100
Subject: [PATCH] Fix L to not broadcast it (#21724)

---
 .../op_conversions/scaled_dot_product_decomposition_test.cpp  | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/common/transformations/tests/op_conversions/scaled_dot_product_decomposition_test.cpp b/src/common/transformations/tests/op_conversions/scaled_dot_product_decomposition_test.cpp
index fec2fa3472be31..92748222eada53 100644
--- a/src/common/transformations/tests/op_conversions/scaled_dot_product_decomposition_test.cpp
+++ b/src/common/transformations/tests/op_conversions/scaled_dot_product_decomposition_test.cpp
@@ -73,7 +73,7 @@ TEST_F(TransformationTestsF, ScaledDotProductAttentionDecompositionStaticBroadca
     const PartialShape query_shape{2, 1, 3, 1};
     const PartialShape key_shape{1, 7, 4, 1};
     const PartialShape value_shape{2, 7, 4, 1};
-    const PartialShape attention_mask_shape{1, 1, 1, 4};
+    const PartialShape attention_mask_shape{1, 1, 3, 4};
     const PartialShape scale_shape{1};
 
     const auto query = std::make_shared<ov::op::v0::Parameter>(element::f32, query_shape);
@@ -100,7 +100,7 @@ TEST_F(TransformationTestsF, ScaledDotProductAttentionDecompositionStaticBroadca
 }
 
 TEST_F(TransformationTestsF, ScaledDotProductAttentionDecompositionStaticBroadcastQuery) {
-    const PartialShape query_shape{1, 7, 1, 4};
+    const PartialShape query_shape{1, 7, 3, 4};
     const PartialShape key_shape{2, 1, 1, 4};
     const PartialShape value_shape{1, 1, 1, 6};
     const PartialShape attention_mask_shape{1, 1, 3, 1};