|
| 1 | +// Copyright (C) 2018-2025 Intel Corporation |
| 2 | +// SPDX-License-Identifier: Apache-2.0 |
| 3 | +// |
| 4 | + |
| 5 | +#include <gmock/gmock.h> |
| 6 | + |
| 7 | +#include <optional> |
| 8 | + |
| 9 | +#include "openvino/core/model.hpp" |
| 10 | +#include "openvino/core/model_util.hpp" |
| 11 | +#include "openvino/op/add.hpp" |
| 12 | +#include "openvino/op/multiply.hpp" |
| 13 | + |
| 14 | +namespace ov::test { |
| 15 | + |
| 16 | +using op::v0::Parameter, op::v1::Add, op::v1::Multiply, op::v0::Result; |
| 17 | + |
| 18 | +class ModelUtilTest : public testing::Test { |
| 19 | +protected: |
| 20 | + static auto make_model_without_tensor_names() { |
| 21 | + auto input_1 = std::make_shared<Parameter>(element::f32, Shape{ 1, 3}); |
| 22 | + auto input_2 = std::make_shared<Parameter>(element::f32, Shape{ 3, 3}); |
| 23 | + auto input_3 = std::make_shared<Parameter>(element::f32, Shape{ 3, 1}); |
| 24 | + |
| 25 | + auto add = std::make_shared<Add>(input_1, input_2); |
| 26 | + auto mul = std::make_shared<Multiply>(add, input_3); |
| 27 | + auto add_final = std::make_shared<Add>(mul, add); |
| 28 | + |
| 29 | + auto add_result = std::make_shared<Result>(add); |
| 30 | + auto mul_result = std::make_shared<Result>(mul); |
| 31 | + auto final_result = std::make_shared<Result>(add_final); |
| 32 | + |
| 33 | + auto model = std::make_shared<Model>(ResultVector{add_result, mul_result, final_result}, |
| 34 | + ParameterVector{input_1, input_2, input_3}, |
| 35 | + "ModelWithoutTensorNames"); |
| 36 | + return model; |
| 37 | + } |
| 38 | + |
| 39 | + static auto make_model_with_named_nodes() { |
| 40 | + auto model = make_model_without_tensor_names(); |
| 41 | + |
| 42 | + const auto& inputs = model->inputs(); |
| 43 | + for (size_t i = 0; i < inputs.size(); ++i) { |
| 44 | + inputs[i].get_node()->set_friendly_name("input_" + std::to_string(i + 1)); |
| 45 | + } |
| 46 | + |
| 47 | + auto results = model->get_results(); |
| 48 | + results[0]->set_friendly_name("add_result"); |
| 49 | + results[1]->set_friendly_name("mul_result"); |
| 50 | + results[2]->set_friendly_name("final_result"); |
| 51 | + model->set_friendly_name("ModelWithNamedNodes"); |
| 52 | + return model; |
| 53 | + } |
| 54 | + |
| 55 | + static auto compare_tensor_names(const OutputVector& outputs, const TensorNamesMap& expected) { |
| 56 | + std::optional<std::string> mismatch_err; |
| 57 | + |
| 58 | + for (const auto& [port, expected_names] : expected) { |
| 59 | + if (const auto& names = outputs[port].get_names(); expected_names != names) { |
| 60 | + using testing::PrintToString; |
| 61 | + mismatch_err.emplace("Tensor names mismatch on port " + PrintToString(port) + "\n Expected: " + |
| 62 | + PrintToString(expected_names) + "\n Actual: " + PrintToString(names)); |
| 63 | + } |
| 64 | + } |
| 65 | + return mismatch_err; |
| 66 | + } |
| 67 | +}; |
| 68 | + |
| 69 | +TEST_F(ModelUtilTest, manual_set_all_input_tensors_names) { |
| 70 | + const auto inputs_names = TensorNamesMap{{0, {"input_1"}}, {1, {"input_2"}}, {2, {"input_3"}}}; |
| 71 | + |
| 72 | + auto model = make_model_without_tensor_names(); |
| 73 | + util::set_input_tensors_names(*model, inputs_names); |
| 74 | + |
| 75 | + ASSERT_EQ(model->inputs().size(), inputs_names.size()); |
| 76 | + for (const auto& [port, names] : inputs_names) { |
| 77 | + EXPECT_EQ(model->input(port).get_names(), names) << "Names not match for input port " << port; |
| 78 | + } |
| 79 | +} |
| 80 | +TEST_F(ModelUtilTest, manual_set_some_input_tensors_names) { |
| 81 | + const auto inputs_names = TensorNamesMap{{2, {"input_2", "mul_input"}}, {0, {"add_input"}}}; |
| 82 | + auto expected_names = inputs_names; |
| 83 | + expected_names.emplace(1, TensorNames{}); |
| 84 | + |
| 85 | + auto model = make_model_without_tensor_names(); |
| 86 | + util::set_input_tensors_names(*model, inputs_names); |
| 87 | + |
| 88 | + ASSERT_EQ(model->inputs().size(), expected_names.size()); |
| 89 | + const auto mismatch_error = compare_tensor_names(model->inputs(), expected_names); |
| 90 | + EXPECT_FALSE(mismatch_error) << *mismatch_error; |
| 91 | +} |
| 92 | + |
| 93 | +TEST_F(ModelUtilTest, auto_set_all_input_tensors_names) { |
| 94 | + const auto expected_names = TensorNamesMap{{0, {"input_1"}}, {1, {"input_2"}}, {2, {"input_3"}}}; |
| 95 | + |
| 96 | + auto model = make_model_with_named_nodes(); |
| 97 | + util::set_input_tensors_names(AUTO, *model); |
| 98 | + |
| 99 | + ASSERT_EQ(model->inputs().size(), expected_names.size()); |
| 100 | + const auto mismatch_error = compare_tensor_names(model->inputs(), expected_names); |
| 101 | + EXPECT_FALSE(mismatch_error) << *mismatch_error; |
| 102 | +} |
| 103 | + |
| 104 | +TEST_F(ModelUtilTest, auto_set_missing_input_tensors_names) { |
| 105 | + const auto inputs_names = TensorNamesMap{{2, {"input_2", "mul_input"}}, {0, {"add_input"}}}; |
| 106 | + auto expected_names = inputs_names; |
| 107 | + expected_names.emplace(1, TensorNames{"input_2"}); |
| 108 | + |
| 109 | + auto model = make_model_with_named_nodes(); |
| 110 | + util::set_input_tensors_names(AUTO, *model, inputs_names); |
| 111 | + |
| 112 | + ASSERT_EQ(model->inputs().size(), expected_names.size()); |
| 113 | + const auto mismatch_error = compare_tensor_names(model->inputs(), expected_names); |
| 114 | + EXPECT_FALSE(mismatch_error) << *mismatch_error; |
| 115 | +} |
| 116 | + |
| 117 | +TEST_F(ModelUtilTest, auto_set_all_io_tensors_names) { |
| 118 | + const auto exp_inputs_names = TensorNamesMap{{2, {"input_3"}}, {0, {"input_1"}}, {1, {"input_2"}}}; |
| 119 | + const auto exp_outputs_names = TensorNamesMap{{0, {"add_result"}}, {1, {"mul_result"}}, {2, {"final_result"}}}; |
| 120 | + |
| 121 | + auto model = make_model_with_named_nodes(); |
| 122 | + util::set_tensors_names(AUTO, *model); |
| 123 | + |
| 124 | + ASSERT_EQ(model->inputs().size(), exp_inputs_names.size()); |
| 125 | + auto mismatch_error = compare_tensor_names(model->inputs(), exp_inputs_names); |
| 126 | + EXPECT_FALSE(mismatch_error) << *mismatch_error; |
| 127 | + |
| 128 | + ASSERT_EQ(model->outputs().size(), exp_outputs_names.size()); |
| 129 | + mismatch_error = compare_tensor_names(model->outputs(), exp_outputs_names); |
| 130 | + EXPECT_FALSE(mismatch_error) << *mismatch_error; |
| 131 | +} |
| 132 | + |
| 133 | +TEST_F(ModelUtilTest, manual_set_all_io_tensors_names) { |
| 134 | + const auto inputs_names = TensorNamesMap{{0, {"input_1"}}, {1, {"input_2"}}, {2, {"input_3"}}}; |
| 135 | + const auto outputs_names = TensorNamesMap{{0, {"add_result"}}, {1, {"mul_result"}}, {2, {"final_result"}}}; |
| 136 | + |
| 137 | + auto model = make_model_without_tensor_names(); |
| 138 | + util::set_tensors_names(*model, inputs_names, outputs_names); |
| 139 | + |
| 140 | + ASSERT_EQ(model->inputs().size(), inputs_names.size()); |
| 141 | + auto mismatch_error = compare_tensor_names(model->inputs(), inputs_names); |
| 142 | + EXPECT_FALSE(mismatch_error) << *mismatch_error; |
| 143 | + |
| 144 | + ASSERT_EQ(model->outputs().size(), outputs_names.size()); |
| 145 | + mismatch_error = compare_tensor_names(model->outputs(), outputs_names); |
| 146 | + EXPECT_FALSE(mismatch_error) << *mismatch_error; |
| 147 | +} |
| 148 | + |
| 149 | +TEST_F(ModelUtilTest, manual_set_some_io_tensors_names) { |
| 150 | + const auto inputs_names = TensorNamesMap{{0, {"input_1"}}, {2, {"input_3"}}}; |
| 151 | + auto expected_input_names = inputs_names; |
| 152 | + expected_input_names.emplace(1, TensorNames{}); |
| 153 | + |
| 154 | + const auto outputs_names = TensorNamesMap{{1, {"mul_result"}}}; |
| 155 | + auto expected_output_names = outputs_names; |
| 156 | + expected_output_names.emplace(0, TensorNames{}); |
| 157 | + expected_output_names.emplace(2, TensorNames{}); |
| 158 | + |
| 159 | + auto model = make_model_without_tensor_names(); |
| 160 | + util::set_tensors_names(*model, inputs_names, outputs_names); |
| 161 | + |
| 162 | + ASSERT_EQ(model->inputs().size(), expected_input_names.size()); |
| 163 | + auto mismatch_error = compare_tensor_names(model->inputs(), expected_input_names); |
| 164 | + EXPECT_FALSE(mismatch_error) << *mismatch_error; |
| 165 | + |
| 166 | + ASSERT_EQ(model->outputs().size(), expected_output_names.size()); |
| 167 | + mismatch_error = compare_tensor_names(model->outputs(), expected_output_names); |
| 168 | + EXPECT_FALSE(mismatch_error) << *mismatch_error; |
| 169 | +} |
| 170 | + |
| 171 | +TEST_F(ModelUtilTest, manual_set_all_output_tensors_names) { |
| 172 | + const auto outputs_names = TensorNamesMap{{0, {"add_result"}}, {1, {"mul_result"}}, {2, {"final_result"}}}; |
| 173 | + |
| 174 | + auto model = make_model_with_named_nodes(); |
| 175 | + util::set_output_tensor_names(*model, outputs_names); |
| 176 | + |
| 177 | + ASSERT_EQ(model->outputs().size(), outputs_names.size()); |
| 178 | + const auto mismatch_error = compare_tensor_names(model->outputs(), outputs_names); |
| 179 | + EXPECT_FALSE(mismatch_error) << *mismatch_error; |
| 180 | +} |
| 181 | + |
| 182 | +TEST_F(ModelUtilTest, manual_set_some_output_tensors_names) { |
| 183 | + const auto outputs_names = TensorNamesMap{{1, {"mul_result"}}, {2, {"final_result"}}}; |
| 184 | + auto expected_names = outputs_names; |
| 185 | + expected_names.emplace(0, TensorNames{}); |
| 186 | + |
| 187 | + auto model = make_model_with_named_nodes(); |
| 188 | + util::set_output_tensor_names(*model, outputs_names); |
| 189 | + |
| 190 | + ASSERT_EQ(model->outputs().size(), expected_names.size()); |
| 191 | + const auto mismatch_error = compare_tensor_names(model->outputs(), expected_names); |
| 192 | + EXPECT_FALSE(mismatch_error) << *mismatch_error; |
| 193 | +} |
| 194 | + |
| 195 | +TEST_F(ModelUtilTest, auto_set_all_output_tensors_names) { |
| 196 | + const auto expected_names = TensorNamesMap{{0, {"add_result"}}, {1, {"mul_result"}}, {2, {"final_result"}}}; |
| 197 | + |
| 198 | + auto model = make_model_with_named_nodes(); |
| 199 | + util::set_output_tensor_names(AUTO, *model); |
| 200 | + |
| 201 | + ASSERT_EQ(model->outputs().size(), expected_names.size()); |
| 202 | + const auto mismatch_error = compare_tensor_names(model->outputs(), expected_names); |
| 203 | + EXPECT_FALSE(mismatch_error) << *mismatch_error; |
| 204 | +} |
| 205 | + |
| 206 | +TEST_F(ModelUtilTest, auto_set_missing_output_tensors_names) { |
| 207 | + const auto outputs_names = TensorNamesMap{{1, {"mul_result"}}, {2, {"final_result"}}}; |
| 208 | + auto expected_names = outputs_names; |
| 209 | + expected_names.emplace(0, TensorNames{"add_result"}); |
| 210 | + |
| 211 | + auto model = make_model_with_named_nodes(); |
| 212 | + util::set_output_tensor_names(AUTO, *model, outputs_names); |
| 213 | + |
| 214 | + ASSERT_EQ(model->outputs().size(), expected_names.size()); |
| 215 | + const auto mismatch_error = compare_tensor_names(model->outputs(), expected_names); |
| 216 | + EXPECT_FALSE(mismatch_error) << *mismatch_error; |
| 217 | +} |
| 218 | + |
| 219 | +} // namespace ov::test |
0 commit comments