diff --git a/tf_adapter/ops/aicpu/npu_cpu_ops.cc b/tf_adapter/ops/aicpu/npu_cpu_ops.cc index 912dfdb45d92713fdd6854ac01943a8578a09735..7dfcc72b3ae43816ec0b7c77d5ff157fd8ee968d 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));