Add f4e2m1 to Java API

openvinotoolkit · Jun 6, 2024 · 898b55c · 898b55c
1 parent 87caec5
commit 898b55c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/modules/java_api/src/main/cpp/jni_common.hpp b/modules/java_api/src/main/cpp/jni_common.hpp
@@ -260,7 +260,8 @@ static const ov::element::Type_t& get_ov_type(int type)
         ov::element::Type_t::nf4,
         ov::element::Type_t::f8e4m3,
         ov::element::Type_t::f8e5m2,
-        ov::element::Type_t::string
+        ov::element::Type_t::string,
+        ov::element::Type_t::f4e2m1
     };
 
     return java_type_to_ov_type.at(type);

diff --git a/modules/java_api/src/main/java/org/intel/openvino/ElementType.java b/modules/java_api/src/main/java/org/intel/openvino/ElementType.java
@@ -31,7 +31,8 @@ public enum ElementType {
     nf4(21),
     f8e4m3(22),
     f8e5m2(23),
-    string(24);
+    string(24),
+    f8e5m2(25);
 
     private int value;
     private static Map<Integer, ElementType> map = new HashMap<Integer, ElementType>();