From c6ee2082025c7f40726ce42d819cf3f9844317f7 Mon Sep 17 00:00:00 2001 From: xumingqian Date: Tue, 13 Aug 2024 14:25:20 +0800 Subject: [PATCH 1/2] kong --- tf_adapter/python/npu_bridge/embedding/embedding_service.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tf_adapter/python/npu_bridge/embedding/embedding_service.py b/tf_adapter/python/npu_bridge/embedding/embedding_service.py index cc1292f77..af9b696fc 100644 --- a/tf_adapter/python/npu_bridge/embedding/embedding_service.py +++ b/tf_adapter/python/npu_bridge/embedding/embedding_service.py @@ -103,6 +103,8 @@ def es_initializer(initializer_mode, min=-2.0, max=2.0, constant_value=0.0, mu=0 seed=seed) + + def check_common_init_params(name, init_vocabulary_size, embedding_dim, embedding_type, mask_zero): if (name is None) or (init_vocabulary_size is None) or (embedding_dim is None): raise ValueError("table name, init_vocabulary_size and embedding_dim can not be None.") -- Gitee From 890360fca00b35cf652ad145e237bd31590a242d Mon Sep 17 00:00:00 2001 From: xumingqian Date: Tue, 13 Aug 2024 14:45:20 +0800 Subject: [PATCH 2/2] kong --- tf_adapter/python/npu_bridge/embedding/embedding_service.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/tf_adapter/python/npu_bridge/embedding/embedding_service.py b/tf_adapter/python/npu_bridge/embedding/embedding_service.py index af9b696fc..cc1292f77 100644 --- a/tf_adapter/python/npu_bridge/embedding/embedding_service.py +++ b/tf_adapter/python/npu_bridge/embedding/embedding_service.py @@ -103,8 +103,6 @@ def es_initializer(initializer_mode, min=-2.0, max=2.0, constant_value=0.0, mu=0 seed=seed) - - def check_common_init_params(name, init_vocabulary_size, embedding_dim, embedding_type, mask_zero): if (name is None) or (init_vocabulary_size is None) or (embedding_dim is None): raise ValueError("table name, init_vocabulary_size and embedding_dim can not be None.") -- Gitee