2 * Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
17 #include "ir/operation/ElementwiseActivation.h"
20 #include <unordered_map>
22 #include "ir/OperationVisitor.h"
31 void ElementwiseActivation::accept(OperationVisitor &v) const { v.visit(*this); }
33 ElementwiseActivation::ElementwiseActivation(const OperandIndexSequence &inputs,
34 const OperandIndexSequence &outputs,
36 : Operation{OperandConstraint::createExact(1u), inputs, outputs}, _param{param}
38 if (param.op_type == Type::LOGISTIC)
40 assert(param.alpha == 0.0f && param.beta == 0.0f && "Logistic will be supported only as "
41 "sigmoid function(L=1, k=1, x0=0). So, do "
42 "not use alpha and beta");
44 else if (param.op_type == Type::RELU)
46 assert(param.alpha >= param.beta && "ReLU's alpha must be equal or greater than beta");
48 else if (param.op_type == Type::TANH)
50 assert(param.alpha == 1.0f && param.beta == 1.0f && "f(x) = alpha * tanh(beta * x), Tanh is "
51 "supported only the values of alpha and "
56 std::string ElementwiseActivation::name() const
58 using ElementwiseActivationType = onert::ir::operation::ElementwiseActivation::Type;
59 static const std::unordered_map<Type, std::string> name_map{
60 {ElementwiseActivationType::ELU, "ELU"},
61 {ElementwiseActivationType::LOGISTIC, "Logistic"},
62 {ElementwiseActivationType::RELU, "ReLU"},
63 {ElementwiseActivationType::TANH, "Tanh"},
64 {ElementwiseActivationType::LEAKY_RELU, "LeakyRelu"}};
65 return name_map.at(_param.op_type);
68 float ElementwiseActivation::infinity = std::numeric_limits<float>::infinity();
70 } // namespace operation