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 "LogSoftMaxLayer.h"
19 #include "OperationUtils.h"
21 #include <cker/operation/LogSoftMax.h>
32 LogSoftMaxLayer::LogSoftMaxLayer() : _input(nullptr), _output(nullptr), _beta(0.0), _axis(0)
37 void LogSoftMaxLayer::PopulateLookupTable(const float kBeta)
39 const float scale = -_input->data_scale() * kBeta;
40 const int32_t max_uint8 = std::numeric_limits<uint8_t>::max();
41 for (int32_t val = 0; val <= max_uint8; ++val)
43 _table[max_uint8 - val] = expf(scale * val);
47 void LogSoftMaxLayer::logsoftmaxFloat32()
49 nnfw::cker::SoftmaxParams op_params;
50 op_params.beta = _beta;
51 op_params.axis = _axis;
52 nnfw::cker::LogSoftmax(op_params, getShape(_input), getBuffer<float>(_input), getShape(_output),
53 getBuffer<float>(_output));
56 void LogSoftMaxLayer::logsoftmaxQuant8()
58 nnfw::cker::SoftmaxParams op_params;
59 op_params.beta = _beta;
60 op_params.axis = _axis;
61 op_params.table = _table;
62 op_params.zero_point = _output->data_zero_point();
63 op_params.scale = _output->data_scale();
64 nnfw::cker::LogSoftmax(op_params, _input->data_scale(), getShape(_input),
65 getBuffer<uint8_t>(_input), getShape(_output),
66 getBuffer<uint8_t>(_output));
69 void LogSoftMaxLayer::configure(const IPortableTensor *input, const float beta, const int axis,
70 IPortableTensor *output)
76 if (_input->data_type() == OperandType::QUANT_UINT8_ASYMM)
78 PopulateLookupTable(_beta);
82 void LogSoftMaxLayer::run()
84 if (_input->data_type() == OperandType::FLOAT32)
88 else if (_input->data_type() == OperandType::QUANT_UINT8_ASYMM)
94 throw std::runtime_error{"LogSoftmax : unsupported data type"};
100 } // namespace backend