Imported Upstream version 1.25.0
[platform/core/ml/nnfw.git] / compiler / tflchef / core / src / CustomOp / Erf.h
1 /*
2  * Copyright (c) 2023 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 #ifndef __OP_ERF_H__
18 #define __OP_ERF_H__
19
20 #include "OpChef.h"
21
22 class ErfChef final : public OpChef
23 {
24 public:
25   explicit ErfChef(const tflchef::Operation *operation) : _operation{operation}
26   {
27     // DO NOTHING
28   }
29
30 public:
31   tflite::BuiltinOperator code(void) const override { return tflite::BuiltinOperator_CUSTOM; }
32
33   tflite::BuiltinOptions type(void) const override { return tflite::BuiltinOptions_NONE; }
34
35   flatbuffers::Offset<void> value(flatbuffers::FlatBufferBuilder &fbb) const override;
36
37   flatbuffers::Offset<flatbuffers::Vector<uint8_t>>
38   custom_value(flatbuffers::FlatBufferBuilder &fbb) const override;
39
40 private:
41   const tflchef::Operation *_operation;
42 };
43
44 struct ErfChefFactory final : public OpChefFactory
45 {
46   std::unique_ptr<OpChef> create(const tflchef::Operation *operation) const override;
47 };
48
49 #endif // __OP_ERF_H__