From 9bae4c5a53c77d12259c484c37b1572ed9168bc4 Mon Sep 17 00:00:00 2001 From: yanghongtian Date: Wed, 15 Apr 2020 22:07:10 +0800 Subject: [PATCH] update act_op namespace --- lite/kernels/hw_ascend_npu/bridges/act_op.cc | 36 ++++++++------------ 1 file changed, 14 insertions(+), 22 deletions(-) diff --git a/lite/kernels/hw_ascend_npu/bridges/act_op.cc b/lite/kernels/hw_ascend_npu/bridges/act_op.cc index fd8f900f7f..f3905d7ba0 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); -- GitLab