Skip to content

Commit b0b28e3

Browse files
authored
Disable memory sharing for GPU. (openvinotoolkit#21370)
1 parent a9d2dd0 commit b0b28e3

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/bindings/python/src/openvino/frontend/tensorflow/node_decoder.py

+3
Original file line numberDiff line numberDiff line change
@@ -91,6 +91,9 @@ def __init__(self, operation: tf.Operation, share_weights: bool, inner_graph: bo
9191
if not self.m_inner_graph:
9292
variable_value = TFGraphNodeDecoder.get_variable(self.m_operation)
9393
if variable_value is not None:
94+
# Disable sharing for variables which are not on CPU
95+
if "device:CPU" not in variable_value.device:
96+
self.m_shared_memory = False
9497
# does not copy data
9598
self.m_parsed_content = variable_value.__array__()
9699

0 commit comments

Comments
 (0)