322785aebb542146588897076c76442d7675546d
[platform/core/ml/nnfw.git] / runtime / onert / backend / cpu / ops / AbsLayer.cc
1 /*
2  * Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved
3  *
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
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
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.
15  */
16
17 #include "AbsLayer.h"
18
19 #include "OperationUtils.h"
20
21 #include <cker/operation/Elementwise.h>
22
23 namespace onert
24 {
25 namespace backend
26 {
27 namespace cpu
28 {
29 namespace ops
30 {
31
32 AbsLayer::AbsLayer() : _input(nullptr), _output(nullptr)
33 {
34   // DO NOTHING
35 }
36
37 void AbsLayer::absFloat32()
38 {
39   nnfw::cker::Abs(getTensorShape(_input), reinterpret_cast<const float *>(_input->buffer()),
40                   getTensorShape(_output), reinterpret_cast<float *>(_output->buffer()));
41 }
42
43 void AbsLayer::absQuant8() { throw std::runtime_error{"NYI"}; }
44
45 void AbsLayer::configure(const IPortableTensor *input, IPortableTensor *output)
46 {
47   _input = input;
48   _output = output;
49 }
50
51 void AbsLayer::run()
52 {
53   if (_input->data_type() == OperandType::FLOAT32)
54   {
55     absFloat32();
56   }
57   else if (_input->data_type() == OperandType::QUANT_UINT8_ASYMM)
58   {
59     absQuant8();
60   }
61   else
62   {
63     throw std::runtime_error{"Abs: unsupported data type"};
64   }
65 }
66
67 } // namespace ops
68 } // namespace cpu
69 } // namespace backend
70 } // namespace onert