Support object-based programming in eager mode and enable tf.function…

… on dynamic_embedding.
tensorflow · Dec 15, 2021 · 001d90f · 001d90f
1 parent fcbe164
commit 001d90f
Show file tree

Hide file tree

Showing 9 changed files with 1,482 additions and 37 deletions.
diff --git a/tensorflow_recommenders_addons/dynamic_embedding/__init__.py b/tensorflow_recommenders_addons/dynamic_embedding/__init__.py
@@ -38,6 +38,7 @@
     'enable_train_mode',
     'get_model_mode',
     'math',
+    'native_ops',
 ]
 
 from tensorflow_recommenders_addons.dynamic_embedding.python.ops import math_ops as math
@@ -87,5 +88,7 @@
 )
 from tensorflow_recommenders_addons.dynamic_embedding.python.ops.tf_patch import (
     patch_on_tf,)
+from tensorflow_recommenders_addons.dynamic_embedding.python.ops import native_dynamic_embedding_ops as native_ops
+from tensorflow_recommenders_addons.dynamic_embedding.python.ops import layers
 
 patch_on_tf()