Support string dtype for OV backend

openvinotoolkit · Mar 26, 2024 · 01e13dc · 01e13dc
1 parent b7ba5ad
commit 01e13dc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/nncf/common/graph/layer_attributes.py b/nncf/common/graph/layer_attributes.py
@@ -19,6 +19,7 @@
 class Dtype(Enum):
     FLOAT = "float"
     INTEGER = "int"
+    STRING = "string"
 
 
 class BaseLayerAttributes(ABC):

diff --git a/nncf/openvino/graph/nncf_graph_builder.py b/nncf/openvino/graph/nncf_graph_builder.py
@@ -56,6 +56,7 @@ def convert_to_nncf_dtype(ov_dtype: str) -> Dtype:
             "u32": "int",
             "u64": "int",
             "boolean": "int",
+            "string": "string",
         }
         if ov_dtype not in conversion_map:
             raise NotImplementedError(f"NNCF is not yet supported OpenVINO data type: {ov_dtype}.")