Use the shared_memory parameter

openvinotoolkit · Nov 20, 2023 · d401c3d · d401c3d
1 parent 386b2f6
commit d401c3d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nncf/openvino/graph/model_transformer.py b/nncf/openvino/graph/model_transformer.py
@@ -356,7 +356,7 @@ def _set_const_value(node_with_const: ov.Node, const_port_id: int, const_value:
 
         const_shape = const_node.data.shape
         const_value = np.reshape(const_value, const_shape)
-        new_const_node = opset.constant(const_value, dtype=const_node.get_element_type())
+        new_const_node = opset.constant(const_value, dtype=const_node.get_element_type(), shared_memory=True)
         new_const_node.set_friendly_name(const_node.get_friendly_name())
         const_port.replace_source_output(new_const_node.output(0))