From a0001e0740ab54f310755c3c9ab574ef3f7d512c Mon Sep 17 00:00:00 2001 From: liuqiang Date: Wed, 16 Aug 2023 09:56:21 +0800 Subject: [PATCH] es0815 --- tf_adapter/ops/aicpu/npu_cpu_ops.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tf_adapter/ops/aicpu/npu_cpu_ops.cc b/tf_adapter/ops/aicpu/npu_cpu_ops.cc index 912dfdb45..7dfcc72b3 100644 --- a/tf_adapter/ops/aicpu/npu_cpu_ops.cc +++ b/tf_adapter/ops/aicpu/npu_cpu_ops.cc @@ -242,7 +242,7 @@ REGISTER_OP("InitEmbeddingHashmap") .Attr("seed2: int = 0") .Attr("filter_mode: string = 'no_filter' ") .Attr("optimizer_mode: string = '' ") - .Attr("optimizer_params: list(float)") + .Attr("optimizer_params: list(float) = [0.1]") .SetShapeFn(shape_inference::NoOutputs); REGISTER_OP("EmbeddingTableImport") @@ -307,7 +307,7 @@ REGISTER_OP("EmbeddingTableFindAndInit") .Attr("default_key: int = 0") .Attr("default_value: float = 0") .Attr("optimizer_mode: string = '' ") - .Attr("optimizer_params: list(float)") + .Attr("optimizer_params: list(float) = [0.1]") .SetShapeFn([](shape_inference::InferenceContext *c) { ShapeHandle keys_shape; TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &keys_shape)); -- Gitee