Disable memory sharing for GPU. (#21370)

2023-11-30 12:05:25 +01:00 · 2023-11-30 12:05:25 +01:00 · b0b28e3853
commit b0b28e3853
parent a9d2dd0b02
1 changed files with 3 additions and 0 deletions
--- a/src/bindings/python/src/openvino/frontend/tensorflow/node_decoder.py
+++ b/src/bindings/python/src/openvino/frontend/tensorflow/node_decoder.py
@ -91,6 +91,9 @@ class TFGraphNodeDecoder(DecoderBase):
            if not self.m_inner_graph:
                variable_value = TFGraphNodeDecoder.get_variable(self.m_operation)
                if variable_value is not None:
+                    # Disable sharing for variables which are not on CPU
+                    if "device:CPU" not in variable_value.device:
+                        self.m_shared_memory = False
                    # does not copy data
                    self.m_parsed_content = variable_value.__array__()