expose ragged_embedding_ops for safe_embedding_lookup_sparse

tensorflow · May 22, 2024 · 0b9a5ee · 0b9a5ee
1 parent bbce3c7
commit 0b9a5ee
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tensorflow_recommenders_addons/dynamic_embedding/__init__.py b/tensorflow_recommenders_addons/dynamic_embedding/__init__.py
@@ -108,6 +108,7 @@
 from tensorflow_recommenders_addons.dynamic_embedding.python.ops.tf_save_restore_patch import (
     patch_on_tf_save_restore,)
 from tensorflow_recommenders_addons.dynamic_embedding.python.ops import shadow_embedding_ops as shadow_ops
+from tensorflow_recommenders_addons.dynamic_embedding.python.ops import ragged_embedding_ops
 
 patch_on_tf()
 patch_on_tf_save_restore()