From 739dd677129e79ec6d81b65db43a3bf2811e94c1 Mon Sep 17 00:00:00 2001 From: w00520461 Date: Tue, 14 Mar 2023 21:16:57 +0800 Subject: [PATCH 1/2] EmbeddingTableExport add export_mode Attr --- tf_adapter/ops/aicpu/npu_cpu_ops.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/tf_adapter/ops/aicpu/npu_cpu_ops.cc b/tf_adapter/ops/aicpu/npu_cpu_ops.cc index f0f621fb9..a1c677a6c 100644 --- a/tf_adapter/ops/aicpu/npu_cpu_ops.cc +++ b/tf_adapter/ops/aicpu/npu_cpu_ops.cc @@ -344,6 +344,7 @@ REGISTER_OP("EmbeddingTableExport") .Input("table_id: int32") .Attr("embedding_dim: int = 0") .Attr("value_total_len: int = 0") + .Attr("export_mode: {'all', 'old', 'new'} = 'all'") .Attr("only_var_flag: bool = false") .Attr("file_type: string = 'bin' ") .SetShapeFn(shape_inference::NoOutputs); -- Gitee From dd7e2979b8617a8c51479754f8acd0c43205d1de Mon Sep 17 00:00:00 2001 From: w00520461 Date: Wed, 22 Mar 2023 11:37:12 +0800 Subject: [PATCH 2/2] EmbeddingTableExport add export_mode Attr --- tf_adapter/ops/aicpu/npu_cpu_ops.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tf_adapter/ops/aicpu/npu_cpu_ops.cc b/tf_adapter/ops/aicpu/npu_cpu_ops.cc index a1c677a6c..d8294b190 100644 --- a/tf_adapter/ops/aicpu/npu_cpu_ops.cc +++ b/tf_adapter/ops/aicpu/npu_cpu_ops.cc @@ -344,7 +344,7 @@ REGISTER_OP("EmbeddingTableExport") .Input("table_id: int32") .Attr("embedding_dim: int = 0") .Attr("value_total_len: int = 0") - .Attr("export_mode: {'all', 'old', 'new'} = 'all'") + .Attr("export_mode: {'all', 'old', 'new', 'specifiednew'} = 'all'") .Attr("only_var_flag: bool = false") .Attr("file_type: string = 'bin' ") .SetShapeFn(shape_inference::NoOutputs); -- Gitee