From bb666a6ec40547d079495c8e75fbdcb7eec5c210 Mon Sep 17 00:00:00 2001 From: yangyongqiang_huawei Date: Thu, 29 May 2025 03:24:05 +0000 Subject: [PATCH 1/3] =?UTF-8?q?update=20tf=5Fadapter/ops/aicore/npu=5Faico?= =?UTF-8?q?re=5Fops.cc.=20embdding=5Fhash=5Ftable=5Flook=5Fup=5For=5Finser?= =?UTF-8?q?t=E7=AE=97=E5=AD=90=E6=96=B0=E5=A2=9E=E5=B1=9E=E6=80=A7?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: yangyongqiang_huawei --- tf_adapter/ops/aicore/npu_aicore_ops.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tf_adapter/ops/aicore/npu_aicore_ops.cc b/tf_adapter/ops/aicore/npu_aicore_ops.cc index bb5e99e7f..4c568441d 100644 --- a/tf_adapter/ops/aicore/npu_aicore_ops.cc +++ b/tf_adapter/ops/aicore/npu_aicore_ops.cc @@ -527,6 +527,8 @@ REGISTER_OP("EmbeddingHashTableLookupOrInsert") .Attr("default_key_or_value:bool = false") .Attr("default_key: int = 0") .Attr("default_value: float = 0.0") + .Attr("filter_key_flag :bool = false") + .Attr("filter_key: int = -1") .SetIsStateful() .SetShapeFn([](InferenceContext* c) { int64 num = 0; -- Gitee From 9ca866b0b51737d6e2b46be280437ef6d71716cb Mon Sep 17 00:00:00 2001 From: yangyongqiang_huawei Date: Thu, 29 May 2025 03:26:30 +0000 Subject: [PATCH 2/3] update tf_adapter/python/npu_bridge/npu_cpu/npu_cpu_ops.py. Signed-off-by: yangyongqiang_huawei --- tf_adapter/python/npu_bridge/npu_cpu/npu_cpu_ops.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tf_adapter/python/npu_bridge/npu_cpu/npu_cpu_ops.py b/tf_adapter/python/npu_bridge/npu_cpu/npu_cpu_ops.py index e7c5fa61f..96ac42738 100644 --- a/tf_adapter/python/npu_bridge/npu_cpu/npu_cpu_ops.py +++ b/tf_adapter/python/npu_bridge/npu_cpu/npu_cpu_ops.py @@ -35,14 +35,16 @@ gen_npu_cpu_ops = helper.get_gen_ops() # @param default_key_or_value bool 类型 # @param default_key int 类型 # @param default_value float 类型 +# @param filter_key_flag bool 类型 +# @param filter_key int 类型 # @return values float 类型 def embedding_hashtable_lookup_or_insert(table_handle, keys, bucket_size, embedding_dim, filter_mode, filter_freq, - default_key_or_value, default_key, default_value): + default_key_or_value, default_key, default_value, filter_key_flag, filter_key): """ device embedding feature mapping lookup or insert. """ result = gen_npu_cpu_ops.EmbeddingHashTableLookupOrInsert( table_handle=table_handle, keys=keys, bucket_size=bucket_size, embedding_dim=embedding_dim, filter_mode=filter_mode, filter_freq=filter_freq, default_key_or_value=default_key_or_value, - default_key=default_key, default_value=default_value) + default_key=default_key, default_value=default_value, filter_key_flag=filter_key_flag, filter_key=filter_key) return result -- Gitee From 1c1e03e5b07596173d2b7e6bd6277ca19adb0c0b Mon Sep 17 00:00:00 2001 From: yangyongqiang_huawei Date: Thu, 29 May 2025 03:28:43 +0000 Subject: [PATCH 3/3] update tf_adapter/ops/aicore/npu_aicore_ops.cc. Signed-off-by: yangyongqiang_huawei --- tf_adapter/ops/aicore/npu_aicore_ops.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tf_adapter/ops/aicore/npu_aicore_ops.cc b/tf_adapter/ops/aicore/npu_aicore_ops.cc index 4c568441d..91913fd56 100644 --- a/tf_adapter/ops/aicore/npu_aicore_ops.cc +++ b/tf_adapter/ops/aicore/npu_aicore_ops.cc @@ -527,7 +527,7 @@ REGISTER_OP("EmbeddingHashTableLookupOrInsert") .Attr("default_key_or_value:bool = false") .Attr("default_key: int = 0") .Attr("default_value: float = 0.0") - .Attr("filter_key_flag :bool = false") + .Attr("filter_key_flag: bool = false") .Attr("filter_key: int = -1") .SetIsStateful() .SetShapeFn([](InferenceContext* c) { -- Gitee