diff --git a/src/tests/ngraph_helpers/ngraph_functions/include/ngraph_functions/utils/data_utils.hpp b/src/tests/ngraph_helpers/ngraph_functions/include/ngraph_functions/utils/data_utils.hpp index 8e7dc75d6ad..ee7d6ed7184 100644 --- a/src/tests/ngraph_helpers/ngraph_functions/include/ngraph_functions/utils/data_utils.hpp +++ b/src/tests/ngraph_helpers/ngraph_functions/include/ngraph_functions/utils/data_utils.hpp @@ -35,6 +35,12 @@ generateVector(size_t vec_len, res[i] = static_cast(dist(gen)); } return res; + } else if (std::is_same()) { + std::bernoulli_distribution dist; + for (size_t i = 0; i < vec_len; i++) { + res[i] = static_cast(dist(gen)); + } + return res; } else { // chose values between this range to avoid type overrun (e.g. in case of I8 precision) std::uniform_int_distribution dist(static_cast(startFrom), static_cast(upTo));