diff --git a/tf_adapter/ops/aicpu/npu_cpu_ops.cc b/tf_adapter/ops/aicpu/npu_cpu_ops.cc
index f0f621fb9376673548957e14a550f665079b9f3b..d8294b190d1cb9fee0b2350132f22c239d55dad4 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', 'specifiednew'} = 'all'")
   .Attr("only_var_flag: bool = false")
   .Attr("file_type: string = 'bin' ")
   .SetShapeFn(shape_inference::NoOutputs);