diff --git a/modules/nvidia_plugin/tests/unit/op_buffers_extractor_test.cpp b/modules/nvidia_plugin/tests/unit/op_buffers_extractor_test.cpp index 358d5e962..ff8022035 100644 --- a/modules/nvidia_plugin/tests/unit/op_buffers_extractor_test.cpp +++ b/modules/nvidia_plugin/tests/unit/op_buffers_extractor_test.cpp @@ -73,7 +73,7 @@ class OperationBufferExtractorTest : public testing::Test { auto reshape = std::make_shared(add_1, reshape_pattern, true); ov::ParameterVector inputs{input}; - ov::NodeVector outputs{reshape}; + ov::OutputVector outputs{reshape}; model_ = std::make_unique(outputs, inputs, "SimpleGraph"); exec_sequence_ = model_->get_ordered_ops(); @@ -346,7 +346,7 @@ class OperationBufferExtractorConcatOptimizedTest : public testing::Test { auto reshape = std::make_shared(add_1, reshape_pattern, true); ov::ParameterVector inputs{input}; - ov::NodeVector outputs{reshape}; + ov::OutputVector outputs{reshape}; model_ = std::make_unique(outputs, inputs, "ConcatOptimizedGraph"); exec_sequence_ = model_->get_ordered_ops(); @@ -482,7 +482,7 @@ class OperationBufferExtractorConcatOptimizedV2Test : public testing::Test { auto reshape1 = std::make_shared(add_1, reshape_pattern, true); ov::ParameterVector inputs{input}; - ov::NodeVector outputs{reshape1}; + ov::OutputVector outputs{reshape1}; model_ = std::make_unique(outputs, inputs, "ConcatOptimizedGraph"); exec_sequence_ = model_->get_ordered_ops();