From c24140166d2bcaedacd5f3b3e2c128a787e74e0d Mon Sep 17 00:00:00 2001 From: Alexandra Sidorova Date: Thu, 25 May 2023 10:08:36 +0400 Subject: [PATCH] Updated data generation in tests --- src/tests/functional/plugin/shared/src/snippets/mha.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/tests/functional/plugin/shared/src/snippets/mha.cpp b/src/tests/functional/plugin/shared/src/snippets/mha.cpp index 74cf209268a442..877217d85ee1a4 100644 --- a/src/tests/functional/plugin/shared/src/snippets/mha.cpp +++ b/src/tests/functional/plugin/shared/src/snippets/mha.cpp @@ -71,7 +71,7 @@ void MHA::generate_inputs(const std::vector& targetInputStaticSha for (int i = 0; i < model_inputs.size(); ++i) { const auto& model_input = model_inputs[i]; ov::Tensor tensor; - tensor = ov::test::utils::create_and_fill_tensor(model_input.get_element_type(), model_input.get_shape(), 21, -10, 256); + tensor = ov::test::utils::create_and_fill_tensor(model_input.get_element_type(), model_input.get_shape(), 2, -1, 256); inputs.insert({model_input.get_node_shared_ptr(), tensor}); } } @@ -93,7 +93,7 @@ void MHASelect::generate_inputs(const std::vector& targetInputSta tensor = ov::test::utils::create_and_fill_tensor(model_input.get_element_type(), model_input.get_shape(), 5 + seed, -2, 10, seed); seed++; } else { - tensor = ov::test::utils::create_and_fill_tensor_normal_distribution(model_input.get_element_type(), model_input.get_shape(), 1.0f, 0.5f); + tensor = ov::test::utils::create_and_fill_tensor(model_input.get_element_type(), model_input.get_shape(), 2, -1, 256); } inputs.insert({node_input, tensor}); }