2 * Copyright (c) 2018 Samsung Electronics Co., Ltd. All Rights Reserved
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
8 * http://www.apache.org/licenses/LICENSE-2.0
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.
17 #ifndef __INTERNAL_OP_BATCHTOSPACE_ND_H__
18 #define __INTERNAL_OP_BATCHTOSPACE_ND_H__
20 #include "internal/op/Node.h"
30 namespace BatchToSpaceNd
38 int32_t block_size_index;
41 Param(uint32_t inputCount, const uint32_t *inputs, uint32_t outputCount, const uint32_t *outputs);
44 } // namespace BatchToSpaceNd
47 } // namespace internal
55 namespace BatchToSpaceNd
57 class Node final : public op::Node
60 Node(const Param ¶m) : _param(param)
66 virtual ~Node() = default;
69 const Param ¶m(void) const { return _param; }
72 void accept(NodeVisitor &&) const override;
78 } // namespace BatchToSpaceNd
81 } // namespace internal
83 #endif // __INTERNAL_OP_BATCHTOSPACE_Nd_H__