From a1cb16587f1006245da7870e8347485ad6e2214c Mon Sep 17 00:00:00 2001 From: yangyongqiang_huawei Date: Tue, 20 May 2025 09:17:46 +0000 Subject: [PATCH 1/3] =?UTF-8?q?update=20tf=5Fadapter/ops/aicore/npu=5Faico?= =?UTF-8?q?re=5Fops.cc.=20=E4=BF=AE=E6=94=B9embedding=5Fhash=5Ftable=5Floo?= =?UTF-8?q?kup=5For=5Finsert=E7=AE=97=E5=AD=90=E5=8A=9F=E8=83=BD=EF=BC=8C?= =?UTF-8?q?=E5=A2=9E=E5=8A=A0filter=5Fkey=E5=8A=9F=E8=83=BD?= 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 077d83788..dbd8d7bbe 100644 --- a/tf_adapter/ops/aicore/npu_aicore_ops.cc +++ b/tf_adapter/ops/aicore/npu_aicore_ops.cc @@ -144,6 +144,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 11acaee703c4e399f85a0e988e026d5fa6e1e939 Mon Sep 17 00:00:00 2001 From: yangyongqiang_huawei Date: Mon, 26 May 2025 02:25:13 +0000 Subject: [PATCH 2/3] =?UTF-8?q?update=20tf=5Fadapter/ops/aicore/npu=5Faico?= =?UTF-8?q?re=5Fops.cc.=20=E4=BF=AE=E6=94=B9lookup=5For=5Finsert=E7=AE=97?= =?UTF-8?q?=E5=AD=90=E8=BE=93=E5=85=A5=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, 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 dbd8d7bbe..20d8111e6 100644 --- a/tf_adapter/ops/aicore/npu_aicore_ops.cc +++ b/tf_adapter/ops/aicore/npu_aicore_ops.cc @@ -144,7 +144,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 From de55004889892692662c424740c674614371960a Mon Sep 17 00:00:00 2001 From: yangyongqiang_huawei Date: Mon, 26 May 2025 02:33:24 +0000 Subject: [PATCH 3/3] =?UTF-8?q?update=20tf=5Fadapter/python/npu=5Fbridge/n?= =?UTF-8?q?pu=5Fcpu/npu=5Fcpu=5Fops.py.=20lookup=5For=5Finsert=E7=AE=97?= =?UTF-8?q?=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/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 273c67241..2eca18aa5 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 @@ -108,14 +108,16 @@ def dense_image_warp_grad(op, grad): # @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