1 // Tencent is pleased to support the open source community by making ncnn available. 2 // 3 // Copyright (C) 2021 THL A29 Limited, a Tencent company. All rights reserved. 4 // 5 // Licensed under the BSD 3-Clause License (the "License"); you may not use this file except 6 // in compliance with the License. You may obtain a copy of the License at 7 // 8 // https://opensource.org/licenses/BSD-3-Clause 9 // 10 // Unless required by applicable law or agreed to in writing, software distributed 11 // under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR 12 // CONDITIONS OF ANY KIND, either express or implied. See the License for the 13 // specific language governing permissions and limitations under the License. 14 15 #include "pass_level1.h" 16 17 #include "../utils.h" 18 19 namespace pnnx { 20 21 class LeakyReLU : public FuseModulePass 22 { 23 public: match_type_str() const24 const char* match_type_str() const 25 { 26 return "__torch__.torch.nn.modules.activation.LeakyReLU"; 27 } 28 type_str() const29 const char* type_str() const 30 { 31 return "nn.LeakyReLU"; 32 } 33 write(Operator * op,const std::shared_ptr<torch::jit::Graph> & graph) const34 void write(Operator* op, const std::shared_ptr<torch::jit::Graph>& graph) const 35 { 36 const torch::jit::Node* leaky_relu = find_node_by_kind(graph, "aten::leaky_relu"); 37 const torch::jit::Node* leaky_relu_ = find_node_by_kind(graph, "aten::leaky_relu_"); 38 39 if (leaky_relu_) 40 { 41 leaky_relu = leaky_relu_; 42 } 43 44 op->params["negative_slope"] = leaky_relu->namedInput("negative_slope"); 45 } 46 }; 47 48 REGISTER_GLOBAL_PNNX_FUSE_MODULE_PASS(LeakyReLU) 49 50 } // namespace pnnx 51