diff --git a/lite/kernels/hw_ascend_npu/bridges/act_op.cc b/lite/kernels/hw_ascend_npu/bridges/act_op.cc index fd8f900f7f1a42c0fea90bc8362b8721369b8ae3..f3905d7ba0e883cfc696b07006eaccc608d241d1 100644 --- a/lite/kernels/hw_ascend_npu/bridges/act_op.cc +++ b/lite/kernels/hw_ascend_npu/bridges/act_op.cc @@ -93,14 +93,6 @@ int ActConverter(void* ctx, } else if (op_type == "relu6") { float Relu_clipped_coef = 6.f; act_op->set_attr_coef(Relu_clipped_coef); - } else if (op_type == "leaky_relu") { - auto alpha = op_info->GetAttr("alpha"); - act_op->set_attr_negative_slope(alpha); - } else if (op_type == "hard_sigmoid") { - auto slope = op_info->GetAttr("slope"); - auto offset = op_info->GetAttr("offset"); - act_op->set_attr_negative_slope(slope); - act_op->set_attr_coef(offset); } return SUCCESS; } @@ -113,53 +105,53 @@ int ActConverter(void* ctx, REGISTER_SUBGRAPH_BRIDGE( sigmoid, kHWAscendNPU, - paddle::lite::subgraph::hw_ascend_npu::ActConverter); + paddle::lite::subgraph::hw_ascend_npu::ActConverter); REGISTER_SUBGRAPH_BRIDGE( relu, kHWAscendNPU, - paddle::lite::subgraph::hw_ascend_npu::ActConverter); + paddle::lite::subgraph::hw_ascend_npu::ActConverter); REGISTER_SUBGRAPH_BRIDGE( tanh, kHWAscendNPU, - paddle::lite::subgraph::hw_ascend_npu::ActConverter); + paddle::lite::subgraph::hw_ascend_npu::ActConverter); REGISTER_SUBGRAPH_BRIDGE( relu_clipped, kNPU, - paddle::lite::subgraph::hw_ascend_npu::ActConverter); + paddle::lite::subgraph::hw_ascend_npu::ActConverter); REGISTER_SUBGRAPH_BRIDGE( relu6, kHWAscendNPU, - paddle::lite::subgraph::hw_ascend_npu::ActConverter); + paddle::lite::subgraph::hw_ascend_npu::ActConverter); REGISTER_SUBGRAPH_BRIDGE( leaky_relu, kHWAscendNPU, - paddle::lite::subgraph::hw_ascend_npu::ActConverter); + paddle::lite::subgraph::hw_ascend_npu::ActConverter); REGISTER_SUBGRAPH_BRIDGE( abs, kHWAscendNPU, - paddle::lite::subgraph::hw_ascend_npu::ActConverter); + paddle::lite::subgraph::hw_ascend_npu::ActConverter); REGISTER_SUBGRAPH_BRIDGE( softsign, - kNPU, - paddle::lite::subgraph::hw_ascend_npu::ActConverter); + kHWAscendNPU, + paddle::lite::subgraph::hw_ascend_npu::ActConverter); REGISTER_SUBGRAPH_BRIDGE( softplus, kHWAscendNPU, - paddle::lite::subgraph::hw_ascend_npu::ActConverter); + paddle::lite::subgraph::hw_ascend_npu::ActConverter); REGISTER_SUBGRAPH_BRIDGE( hard_sigmoid, kHWAscendNPU, - paddle::lite::subgraph::hw_ascend_npu::ActConverter); + paddle::lite::subgraph::hw_ascend_npu::ActConverter); REGISTER_SUBGRAPH_BRIDGE( log, kHWAscendNPU, - paddle::lite::subgraph::hw_ascend_npu::ActConverter); + paddle::lite::subgraph::hw_ascend_npu::ActConverter); REGISTER_SUBGRAPH_BRIDGE( square, kHWAscendNPU, - paddle::lite::subgraph::hw_ascend_npu::ActConverter); + paddle::lite::subgraph::hw_ascend_npu::ActConverter); REGISTER_SUBGRAPH_BRIDGE( sqrt, kHWAscendNPU, - paddle::lite::subgraph::hw_ascend_npu::ActConverter); + paddle::lite::subgraph::hw_ascend_npu::ActConverter);