Imported Upstream version 1.18.0
[platform/core/ml/nnfw.git] / compiler / luci-interpreter / src / loader / nodes / Equal.cpp
1 /*
2  * Copyright (c) 2021 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 "Builders.h"
18
19 #include "kernels/Equal.h"
20
21 namespace luci_interpreter
22 {
23
24 std::unique_ptr<Kernel> build_kernel_CircleEqual(const luci::CircleNode *circle_node,
25                                                  KernelBuilderHelper &helper)
26
27 {
28   const auto *node = dynamic_cast<const luci::CircleEqual *>(circle_node);
29   if (node == nullptr)
30     throw std::runtime_error("wrong builder for operation");
31   assert(node->arity() == 2);
32
33   const Tensor *x = helper.getInputTensor(node->x());
34   const Tensor *y = helper.getInputTensor(node->y());
35   Tensor *output = helper.getOutputTensor(node);
36
37   return std::make_unique<kernels::Equal>(x, y, output);
38 }
39
40 } // namespace luci_interpreter