12640ca0d4a1f7d669840ca9ac202b6360d25e65
[platform/core/ml/nnfw.git] / onert-micro / luci-interpreter / src / kernels / SpaceToBatchND.h
1 /*
2  * Copyright (c) 2021 Samsung Electronics Co., Ltd. All Rights Reserved
3  * Copyright 2019 The TensorFlow Authors. All Rights Reserved.
4  *
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at
8  *
9  *    http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  */
17
18 #ifndef LUCI_INTERPRETER_KERNELS_SPACETOBATCHND_H
19 #define LUCI_INTERPRETER_KERNELS_SPACETOBATCHND_H
20
21 #include "core/Kernel.h"
22
23 namespace luci_interpreter
24 {
25 namespace kernels
26 {
27
28 class SpaceToBatchND : public Kernel
29 {
30 public:
31   SpaceToBatchND(const Tensor *input, const Tensor *block_shape, const Tensor *paddings,
32                  Tensor *output);
33
34   const Tensor *input() const { return _inputs[0]; }
35   const Tensor *block_shape() const { return _inputs[1]; }
36   const Tensor *paddings() const { return _inputs[2]; }
37   Tensor *output() const { return _outputs[0]; }
38
39   void configure() override;
40   void execute() const override;
41 };
42
43 } // namespace kernels
44 } // namespace luci_interpreter
45
46 #endif // LUCI_INTERPRETER_KERNELS_SPACETOBATCHND_H