add custom op demo

alibaba · Oct 21, 2023 · 03b9868 · 03b9868
1 parent efa5683
commit 03b9868
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 1 deletion.
diff --git a/docs/source/component/custom_op.md b/docs/source/component/custom_op.md
@@ -95,6 +95,12 @@ model_config: {
       }
       keras_layer {
         class_name: 'EditDistance'
+        st_params {
+          fields {
+            key: 'text_encoding'
+            value: { string_value: 'latin' }
+          }
+        }
       }
     }
     blocks {

diff --git a/easy_rec/python/layers/keras/custom_ops.py b/easy_rec/python/layers/keras/custom_ops.py
@@ -10,7 +10,7 @@
 curr_dir, _ = os.path.split(__file__)
 parent_dir = os.path.dirname(curr_dir)
 ops_idr = os.path.dirname(parent_dir)
-ops_dir = os.path.join(ops_idr, 'python', 'ops')
+ops_dir = os.path.join(ops_idr, 'ops')
 if 'PAI' in tf.__version__:
   ops_dir = os.path.join(ops_dir, '1.12_pai')
 elif tf.__version__.startswith('1.12'):

diff --git a/easy_rec/python/ops/1.12/libedit_distance.so b/easy_rec/python/ops/1.12/libedit_distance.so
diff --git a/easy_rec/python/ops/1.15/libedit_distance.so b/easy_rec/python/ops/1.15/libedit_distance.so