Performance enhancements for sparse embedding lookups.

tensorflow · Feb 16, 2023 · 2166e5d · 2166e5d
1 parent 497f9c3
commit 2166e5d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/RELEASE.md b/RELEASE.md
@@ -34,6 +34,14 @@
         model and exports it if the user-defined comparison function determines
         that it is an improvement.
 
+*   Embeddings:
+
+    *   `tf.nn.embedding_lookup_sparse` and `tf.nn.safe_embedding_lookup_sparse`
+        now support ids and weights described by RaggedTensors, which are more
+        memory efficient than SparseTensors. The new boolean argument
+        `allow_fast_lookup` enables a simplified and typically faster lookup
+        procedure.
+
 ## Bug Fixes and Other Changes
 
 * <SIMILAR TO ABOVE SECTION, BUT FOR OTHER IMPORTANT CHANGES / BUG FIXES>