From 541af4bca307243e1f25e5e2352200081b5228e0 Mon Sep 17 00:00:00 2001 From: liuqiang Date: Wed, 27 Mar 2024 11:45:38 +0800 Subject: [PATCH] es_0327 --- tf_adapter/python/npu_bridge/embedding/embedding_service.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tf_adapter/python/npu_bridge/embedding/embedding_service.py b/tf_adapter/python/npu_bridge/embedding/embedding_service.py index ef03bae87..712b02f8e 100644 --- a/tf_adapter/python/npu_bridge/embedding/embedding_service.py +++ b/tf_adapter/python/npu_bridge/embedding/embedding_service.py @@ -1793,7 +1793,8 @@ class ESWorker: gen_npu_cpu_ops.embedding_feature_mapping_import(file_path=path, table_name=x[:-2], feature_size=feature_size, - embedding_dim=self._small_table_embedding_dim) + embedding_dim=self._small_table_embedding_dim, + only_offset_flag=True) feature_mapping_insert = \ gen_npu_cpu_ops.embedding_feature_mapping_insert(table_name=x[:-2], feature_id=feature_id, -- Gitee