diff --git a/include/tim/vx/types.h b/include/tim/vx/types.h
index 1937e68..432ff1c 100644
--- a/include/tim/vx/types.h
+++ b/include/tim/vx/types.h
@@ -29,6 +29,8 @@ namespace vx {
 
 enum class DataType {
   UNKNOWN,
+  INT4,
+  UINT4,
   INT8,
   UINT8,
   INT16,
diff --git a/src/tim/vx/type_utils.cc b/src/tim/vx/type_utils.cc
index 11d5681..a701452 100644
--- a/src/tim/vx/type_utils.cc
+++ b/src/tim/vx/type_utils.cc
@@ -27,6 +27,10 @@ namespace tim {
 namespace vx {
 vsi_nn_type_e TranslateDataType(DataType dtype) {
   switch (dtype) {
+    case DataType::INT4:
+        return VSI_NN_TYPE_INT4;
+    case DataType::UINT4:
+        return VSI_NN_TYPE_UINT4;
     case DataType::INT8:
       return VSI_NN_TYPE_INT8;
     case DataType::UINT8: