From 0c2308506fdf8fc616beeb10ed8977a09bc6071d Mon Sep 17 00:00:00 2001 From: Roman Kazantsev Date: Wed, 29 Mar 2023 16:28:37 +0400 Subject: [PATCH] [TF FE] Fix leftovers from review (#16619) Signed-off-by: Kazantsev, Roman --- src/frontends/tensorflow/src/graph_iterator_proto_txt.hpp | 1 + tests/layer_tests/tensorflow_tests/test_tf_Bucketize.py | 1 + 2 files changed, 2 insertions(+) diff --git a/src/frontends/tensorflow/src/graph_iterator_proto_txt.hpp b/src/frontends/tensorflow/src/graph_iterator_proto_txt.hpp index a38b6031e52..b193aaaef52 100644 --- a/src/frontends/tensorflow/src/graph_iterator_proto_txt.hpp +++ b/src/frontends/tensorflow/src/graph_iterator_proto_txt.hpp @@ -22,6 +22,7 @@ public: std::ifstream pbtxt_stream(path, std::ios::in); FRONT_END_GENERAL_CHECK(pbtxt_stream && pbtxt_stream.is_open(), "Model file does not exist"); auto input_stream = std::make_shared<::google::protobuf::io::IstreamInputStream>(&pbtxt_stream); + FRONT_END_GENERAL_CHECK(input_stream, "Model cannot be read"); auto is_parsed = ::google::protobuf::TextFormat::Parse(input_stream.get(), m_graph_def.get()); FRONT_END_GENERAL_CHECK( is_parsed, diff --git a/tests/layer_tests/tensorflow_tests/test_tf_Bucketize.py b/tests/layer_tests/tensorflow_tests/test_tf_Bucketize.py index bf0c2ef7bc3..d981b299754 100644 --- a/tests/layer_tests/tensorflow_tests/test_tf_Bucketize.py +++ b/tests/layer_tests/tensorflow_tests/test_tf_Bucketize.py @@ -23,6 +23,7 @@ class TestBucketize(CommonTFLayerTest): with tf.compat.v1.Session() as sess: input = tf.compat.v1.placeholder(input_type, input_shape, 'input') # generate boundaries list + # use wider range for boundaries than input data in order to cover all bucket indices cases boundaries = np.sort(np.unique(np.random.randint(-200, 200, [boundaries_size]).astype(np.float32))).tolist() tf.raw_ops.Bucketize(input=input, boundaries=boundaries) tf.compat.v1.global_variables_initializer()