Fix L to not broadcast it (openvinotoolkit#21724)

vishniakov-nikolai · Dec 18, 2023 · ec8765f · ec8765f
1 parent 032ac89
commit ec8765f
Showing 1 changed file with 2 additions and 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
@@ -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};