From 7239f8201bf18d953298966afd9161cff50b2d38 Mon Sep 17 00:00:00 2001 From: Pawel Raasz Date: Mon, 10 Jun 2024 21:05:28 +0200 Subject: [PATCH] Use OV API to set tensor descriptor in output for (#920) ParameterStubNode --- modules/nvidia_plugin/tests/unit/result.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/nvidia_plugin/tests/unit/result.cpp b/modules/nvidia_plugin/tests/unit/result.cpp index 105af4beb..33ec8b8dd 100644 --- a/modules/nvidia_plugin/tests/unit/result.cpp +++ b/modules/nvidia_plugin/tests/unit/result.cpp @@ -28,7 +28,7 @@ class ov::Output : public ov::Output { ov::element::Type{}, ov::PartialShape{1}, std::unordered_set{ParameterStubNode::get_type_info_static().name}); - node->m_outputs.emplace_back(node.get(), 0, tensor); + node->get_output_descriptor(0).set_tensor_ptr(tensor); } };