Adds performance enhancements for sparse embedding lookups.

tensorflow · Feb 16, 2023 · 8826a9b · 8826a9b
1 parent 47157c3
commit 8826a9b
Show file tree

Hide file tree

Showing 5 changed files with 391 additions and 283 deletions.
diff --git a/tensorflow/python/eager/benchmarks_test.py b/tensorflow/python/eager/benchmarks_test.py
@@ -58,6 +58,7 @@
 from tensorflow.python.ops import random_ops
 from tensorflow.python.ops import resource_variable_ops
 from tensorflow.python.ops.ragged import ragged_tensor
+from tensorflow.python.ops.ragged import ragged_embedding_ops
 from tensorflow.python.util import nest
 from tensorflow.python.util import tf_inspect