Imported Upstream version 1.25.0
[platform/core/ml/nnfw.git] / onert-micro / luci-interpreter / src / kernels / BatchToSpaceND.h
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 #ifndef LUCI_INTERPRETER_KERNELS_BATCHTOSPACEND_H
18 #define LUCI_INTERPRETER_KERNELS_BATCHTOSPACEND_H
19
20 #include "core/Kernel.h"
21
22 namespace luci_interpreter
23 {
24 namespace kernels
25 {
26
27 class BatchToSpaceND : public Kernel
28 {
29 public:
30   BatchToSpaceND(const Tensor *input, const Tensor *block_shape, const Tensor *crops,
31                  Tensor *output);
32
33   const Tensor *input() const { return _inputs[0]; }
34   const Tensor *block_shape() const { return _inputs[1]; }
35   const Tensor *crops() const { return _inputs[2]; }
36   Tensor *output() const { return _outputs[0]; }
37
38   void configure() override;
39   void execute() const override;
40 };
41
42 } // namespace kernels
43 } // namespace luci_interpreter
44
45 #endif // LUCI_INTERPRETER_KERNELS_BATCHTOSPACEND_H