From e8fcf6540852fa4a31e77f80430198b47ace6f79 Mon Sep 17 00:00:00 2001 From: lovline Date: Mon, 18 Dec 2023 14:36:28 +0800 Subject: [PATCH] host unique counter filter adaption --- 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 266b9d988..7bce51842 100644 --- a/tf_adapter/ops/aicpu/npu_cpu_ops.cc +++ b/tf_adapter/ops/aicpu/npu_cpu_ops.cc @@ -325,6 +325,7 @@ REGISTER_OP("FakeRemoteLookupUniqued") .Input("table_id: int32") .Input("keys: int64") .Input("actual_keys_input: int64") + .Input("key_count: int64") .Input("unique_indices: int32") .Output("values: float32") .Attr("embedding_dim: int = 0") -- Gitee