Add EMBEDDING_LOOKUP to weightonly_wi4_afp32 policy.

PiperOrigin-RevId: 684143235
google-ai-edge · Oct 9, 2024 · 70ad7d3 · 70ad7d3
1 parent 4f7041f
commit 70ad7d3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ai_edge_quantizer/default_policy.py b/ai_edge_quantizer/default_policy.py
@@ -208,7 +208,7 @@
       "EMBEDDING_LOOKUP",
       "FULLY_CONNECTED"
     ],
-    "weightonly_wi4_afp32": ["BATCH_MATMUL", "FULLY_CONNECTED"]
+    "weightonly_wi4_afp32": ["BATCH_MATMUL", "FULLY_CONNECTED", "EMBEDDING_LOOKUP"]
   }
 }
 """