From 68a82e50705b6d4236ae35cc56ad944b3e589fae Mon Sep 17 00:00:00 2001 From: lovline Date: Thu, 10 Aug 2023 09:01:15 +0800 Subject: [PATCH] add filter_export_flag to export op for counter filter --- 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 eba2b48e2..dfacbf76b 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") -- Gitee