diff --git a/tf_adapter/ops/aicpu/npu_cpu_ops.cc b/tf_adapter/ops/aicpu/npu_cpu_ops.cc index f0f621fb9376673548957e14a550f665079b9f3b..9473e352d15b2b365a5d6e9bc7bac26526bb081e 100644 --- a/tf_adapter/ops/aicpu/npu_cpu_ops.cc +++ b/tf_adapter/ops/aicpu/npu_cpu_ops.cc @@ -348,6 +348,7 @@ REGISTER_OP("EmbeddingTableExport") .Attr("file_type: string = 'bin' ") .SetShapeFn(shape_inference::NoOutputs); + REGISTER_OP("EmbeddingFeatureMapping") .Input("feature_id: int64") .Output("offset_id: int32")