diff options
Diffstat (limited to 'runtimes/pure_arm_compute/src/internal/layers/SimpleSpaceToDepth.h')
-rw-r--r-- | runtimes/pure_arm_compute/src/internal/layers/SimpleSpaceToDepth.h | 64 |
1 files changed, 0 insertions, 64 deletions
diff --git a/runtimes/pure_arm_compute/src/internal/layers/SimpleSpaceToDepth.h b/runtimes/pure_arm_compute/src/internal/layers/SimpleSpaceToDepth.h deleted file mode 100644 index 9e87c364c..000000000 --- a/runtimes/pure_arm_compute/src/internal/layers/SimpleSpaceToDepth.h +++ /dev/null @@ -1,64 +0,0 @@ -/* - * Copyright (c) 2018 Samsung Electronics Co., Ltd. All Rights Reserved - * - * Licensed under the Apache License, Version 2.0 (the "License"); - * you may not use this file except in compliance with the License. - * You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - * See the License for the specific language governing permissions and - * limitations under the License. - */ - -/** - * @file SimpleSpaceToDepth.h - * @brief This file contains SimpleSpaceToDepth class - * @ingroup COM_AI_RUNTIME - */ - -#ifndef __SIMPLE_SPACE_TO_DEPTH_H__ -#define __SIMPLE_SPACE_TO_DEPTH_H__ - -#include "internal/arm_compute.h" -#include "internal/arm_compute/Cast.h" - -/** - * @brief Class to run SimpleEmbeddingLookup Layer - */ -class SimpleSpaceToDepth : public ::arm_compute::IFunction -{ -public: - SimpleSpaceToDepth(void) : _input(nullptr), _output(nullptr), _block_size(0), _axises{} - { - // DO NOTHING - } - - /** - * @brief Configure the layer - * @param[in] input First tensor input. - * @param[in] output Output tensor. - * @param[in] block_size Block size. - * @param[in] axises Axises of rank 4 - * @return N/A - */ - void configure(::arm_compute::ITensor *input, ::arm_compute::ITensor *output, int32_t block_size, - const ::arm_compute::Coordinates &axises = getARMComputeAxises(4)); - - /** - * @brief Run the operation. Must be called after configure(). - * @return N/A - */ - void run() override; - -private: - ::arm_compute::ITensor *_input; - ::arm_compute::ITensor *_output; - int32_t _block_size; - ::arm_compute::Coordinates _axises; -}; - -#endif /*__SIMPLE_SPACE_TO_DEPTH_H__ */ |