Add INT4 Conv to supported policy

PiperOrigin-RevId: 682465087
google-ai-edge · Oct 11, 2024 · 1521e82 · 1521e82
1 parent 70ad7d3
commit 1521e82
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ai_edge_quantizer/default_policy.py b/ai_edge_quantizer/default_policy.py
@@ -199,7 +199,7 @@
       "EMBEDDING_LOOKUP",
       "FULLY_CONNECTED"
     ],
-    "dynamic_wi4_afp32": ["FULLY_CONNECTED", "EMBEDDING_LOOKUP"],
+    "dynamic_wi4_afp32": ["FULLY_CONNECTED", "EMBEDDING_LOOKUP", "CONV_2D"],
     "weightonly_wi8_afp32": [
       "BATCH_MATMUL",
       "CONV_2D",
@@ -208,7 +208,7 @@
       "EMBEDDING_LOOKUP",
       "FULLY_CONNECTED"
     ],
-    "weightonly_wi4_afp32": ["BATCH_MATMUL", "FULLY_CONNECTED", "EMBEDDING_LOOKUP"]
+    "weightonly_wi4_afp32": ["BATCH_MATMUL", "FULLY_CONNECTED", "EMBEDDING_LOOKUP", "CONV_2D"]
   }
 }
 """