Use OV API to set tensor descriptor in output for (#920)

ParameterStubNode
openvinotoolkit · Jun 10, 2024 · 7239f82 · 7239f82
1 parent 659e519
commit 7239f82
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/nvidia_plugin/tests/unit/result.cpp b/modules/nvidia_plugin/tests/unit/result.cpp
@@ -28,7 +28,7 @@ class ov::Output<ParameterStubNode> : public ov::Output<ov::Node> {
             ov::element::Type{},
             ov::PartialShape{1},
             std::unordered_set<std::string>{ParameterStubNode::get_type_info_static().name});
-        node->m_outputs.emplace_back(node.get(), 0, tensor);
+        node->get_output_descriptor(0).set_tensor_ptr(tensor);
     }
 };