From 3761d50ca7b7a06addd8cc48b21351a07bfb48c6 Mon Sep 17 00:00:00 2001 From: xumingqian Date: Wed, 31 Jul 2024 10:35:51 +0800 Subject: [PATCH] fix diff dim --- 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 f16998687..05126f931 100644 --- a/tf_adapter/python/npu_bridge/embedding/embedding_service.py +++ b/tf_adapter/python/npu_bridge/embedding/embedding_service.py @@ -1487,7 +1487,7 @@ class ESWorker: if x.name[3:] in self._small_table_variable_list: idx = self._small_table_variable_list.index(x.name[3:]) offset_list[idx] = tf.gather(x, offset_id[idx]) - values = tf.concat(offset_list, axis=0) + values = tf.concat(offset_list, axis=1) else: values = 0 feature_mapping_export = gen_npu_cpu_ops.embedding_feature_mapping_export(file_path=path, -- Gitee