diff --git a/tf_adapter/ops/aicpu/npu_cpu_ops.cc b/tf_adapter/ops/aicpu/npu_cpu_ops.cc index eba2b48e2fd14c6e146a580431c6307b1bdecbb9..dfacbf76b7605f866ac84a75f593fe2ed0c3ed6e 100644 --- a/tf_adapter/ops/aicpu/npu_cpu_ops.cc +++ b/tf_adapter/ops/aicpu/npu_cpu_ops.cc @@ -400,6 +400,7 @@ REGISTER_OP("EmbeddingTableExport") .Attr("only_var_flag: bool = false") .Attr("file_type: string = 'bin' ") .Attr("table_name: list(string)") + .Attr("filter_export_flag: bool = false") .SetShapeFn(shape_inference::NoOutputs); REGISTER_OP("EmbeddingFeatureMapping")