Skip to content

Commit 898b55c

Browse files
committed
Add f4e2m1 to Java API
1 parent 87caec5 commit 898b55c

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

modules/java_api/src/main/cpp/jni_common.hpp

+2-1
Original file line numberDiff line numberDiff line change
@@ -260,7 +260,8 @@ static const ov::element::Type_t& get_ov_type(int type)
260260
ov::element::Type_t::nf4,
261261
ov::element::Type_t::f8e4m3,
262262
ov::element::Type_t::f8e5m2,
263-
ov::element::Type_t::string
263+
ov::element::Type_t::string,
264+
ov::element::Type_t::f4e2m1
264265
};
265266

266267
return java_type_to_ov_type.at(type);

modules/java_api/src/main/java/org/intel/openvino/ElementType.java

+2-1
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,8 @@ public enum ElementType {
3131
nf4(21),
3232
f8e4m3(22),
3333
f8e5m2(23),
34-
string(24);
34+
string(24),
35+
f8e5m2(25);
3536

3637
private int value;
3738
private static Map<Integer, ElementType> map = new HashMap<Integer, ElementType>();

0 commit comments

Comments
 (0)