diff options
Diffstat (limited to 'runtimes/pure_arm_compute/src/internal/layers/SimpleUnpackLayer.h')
-rw-r--r-- | runtimes/pure_arm_compute/src/internal/layers/SimpleUnpackLayer.h | 51 |
1 files changed, 0 insertions, 51 deletions
diff --git a/runtimes/pure_arm_compute/src/internal/layers/SimpleUnpackLayer.h b/runtimes/pure_arm_compute/src/internal/layers/SimpleUnpackLayer.h deleted file mode 100644 index 52fc7513d..000000000 --- a/runtimes/pure_arm_compute/src/internal/layers/SimpleUnpackLayer.h +++ /dev/null @@ -1,51 +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. - */ -#ifndef __UNPACK_LAYER_H__ -#define __UNPACK_LAYER_H__ - -#include <arm_compute/runtime/CL/CLTensor.h> -#include <arm_compute/runtime/CL/CLSubTensor.h> -#include <arm_compute/runtime/CL/functions/CLReshapeLayer.h> -#include <arm_compute/runtime/CL/functions/CLPermute.h> - -class SimpleUnpackLayer : public ::arm_compute::IFunction -{ -public: - SimpleUnpackLayer(void) - : _cl_permuted_vector{}, _output_vector{}, _sub_tensor_vector{}, _cl_reshape_vector{}, - _cl_permute_vector{}, _input(nullptr), _axis(0) - { - // DO NOTHING - } - -public: - void configure(::arm_compute::ICLTensor *input, - const std::vector<::arm_compute::ICLTensor *> &output_vector, int32_t axis); - -public: - void run(void) override; - -private: - std::vector<::arm_compute::CLTensor> _cl_permuted_vector; - std::vector<::arm_compute::ICLTensor *> _output_vector; - std::vector<std::shared_ptr<::arm_compute::CLSubTensor>> _sub_tensor_vector; - std::vector<::arm_compute::CLReshapeLayer> _cl_reshape_vector; - std::vector<::arm_compute::CLPermute> _cl_permute_vector; - ::arm_compute::ICLTensor *_input; - int32_t _axis; -}; - -#endif // __UNPACK_LAYER_H__ |