diff options
Diffstat (limited to 'runtimes/pure_arm_compute/src/internal/layers/SimpleArgMinMax.h')
-rw-r--r-- | runtimes/pure_arm_compute/src/internal/layers/SimpleArgMinMax.h | 51 |
1 files changed, 0 insertions, 51 deletions
diff --git a/runtimes/pure_arm_compute/src/internal/layers/SimpleArgMinMax.h b/runtimes/pure_arm_compute/src/internal/layers/SimpleArgMinMax.h deleted file mode 100644 index b90e74579..000000000 --- a/runtimes/pure_arm_compute/src/internal/layers/SimpleArgMinMax.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 __SIMPLE_ARG_MIN_MAX_H__ -#define __SIMPLE_ARG_MIN_MAX_H__ - -#include "internal/arm_compute.h" -#include "arm_compute/core/TypesEx.h" - -class SimpleArgMinMax : public ::arm_compute::IFunction -{ -public: - SimpleArgMinMax(void) : _input(nullptr), _output(nullptr), _axis(), _input_rank(0) - { - // DO NOTHING - } - -public: - /** Initialise input and output - * - * @param[in] input First tensor input. - * @param[out] output Output tensor. - * @param[in] axis Dimension along which to find Min or Max Index. - */ - void configure(::arm_compute::ITensor *input, ::arm_compute::ITensor *output, - std::vector<uint32_t> axis, ::arm_compute::ArgOperation _op_type); - - void run() override; - -private: - ::arm_compute::ITensor *_input; - ::arm_compute::ITensor *_output; - std::vector<uint32_t> _axis; - int _input_rank; - ::arm_compute::ArgOperation _op_type; -}; - -#endif /*__SIMPLE_ARG_MIN_MAX_H__ */ |