From 01393b07d2cce6e36e4b477f849a481fdbe39994 Mon Sep 17 00:00:00 2001 From: liuqiang Date: Sun, 29 Jan 2023 17:04:45 +0800 Subject: [PATCH] xiugai2 --- tf_adapter/python/npu_bridge/embedding/embedding_service.py | 2 +- 1 file changed, 1 insertion(+), 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 fe6c9803e..5b5e075cb 100644 --- a/tf_adapter/python/npu_bridge/embedding/embedding_service.py +++ b/tf_adapter/python/npu_bridge/embedding/embedding_service.py @@ -241,7 +241,7 @@ class ESWorker: ps_id=ops.convert_to_tensor(-1), table_id=ops.convert_to_tensor(table_id), embedding_dim=embedding_dim, - value_total_len=embedding_dim * (self.slot_vars_num + 1), + value_total_len=embedding_dim, only_var_flag=only_var, file_type=mode) return tf.group([embedding_table_import]) -- Gitee