From e222f726e0f058cdcf75cc253b56029990ce593a Mon Sep 17 00:00:00 2001 From: "Raasz, Pawel" Date: Mon, 3 Jun 2024 06:08:03 +0000 Subject: [PATCH] Use OV API to set tensor descriptor in output for 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); } };