diff options
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEReshapeLayer.h')
-rw-r--r-- | arm_compute/runtime/NEON/functions/NEReshapeLayer.h | 51 |
1 files changed, 46 insertions, 5 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEReshapeLayer.h b/arm_compute/runtime/NEON/functions/NEReshapeLayer.h index d6643842d9..5a296a776d 100644 --- a/arm_compute/runtime/NEON/functions/NEReshapeLayer.h +++ b/arm_compute/runtime/NEON/functions/NEReshapeLayer.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2019 ARM Limited. + * Copyright (c) 2017-2020 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -24,8 +24,11 @@ #ifndef ARM_COMPUTE_NERESHAPELAYER_H #define ARM_COMPUTE_NERESHAPELAYER_H +#include "arm_compute/core/NEON/kernels/NEReshapeLayerKernel.h" #include "arm_compute/core/Types.h" -#include "arm_compute/runtime/NEON/INESimpleFunctionNoBorder.h" +#include "arm_compute/runtime/IFunction.h" +#include "arm_compute/runtime/NEON/INEOperator.h" +#include "arm_compute/runtime/Types.h" namespace arm_compute { @@ -33,24 +36,62 @@ namespace arm_compute class ITensor; /** Basic function to run @ref NEReshapeLayerKernel */ -class NEReshapeLayer : public INESimpleFunctionNoBorder +class NEReshapeLayer : public IFunction { public: /** Initialise the kernel's inputs and outputs * - * @param[in] input First tensor input. Data type supported: All + * @param[in] input Input tensor. Data type supported: All * @param[out] output Output tensor. Data type supported: Same as @p input */ void configure(const ITensor *input, ITensor *output); /** Static function to check if given info will lead to a valid configuration of @ref NEReshapeLayer * - * @param[in] input First tensor info. Data type supported: All + * @param[in] input Input tensor info. Data type supported: All * @param[in] output Output tensor info. Data type supported: Same as @p input * * @return a status */ static Status validate(const ITensorInfo *input, const ITensorInfo *output); + + // Inherited methods overridden: + void run() override; + +private: + const ITensor *_input + { + nullptr + }; + ITensor *_output{ nullptr }; + std::unique_ptr<NEReshapeLayerKernel> _kernel{ nullptr }; +}; + +namespace experimental +{ +/** Basic function to run @ref NEReshapeLayerKernel */ +class NEReshapeLayer : public INEOperator +{ +public: + /** Initialise the kernel's inputs and outputs + * + * @param[in] input Input tensor info. Data type supported: All + * @param[out] output Output info. Data type supported: Same as @p input + */ + void configure(const ITensorInfo *input, ITensorInfo *output); + + /** Static function to check if given info will lead to a valid configuration of @ref NEReshapeLayer + * + * @param[in] input Input tensor info. Data type supported: All + * @param[in] output Output tensor info. Data type supported: Same as @p input + * + * @return a status + */ + static Status validate(const ITensorInfo *input, const ITensorInfo *output); + + // Inherited methods overridden: + MemoryRequirements workspace() const override; }; +} // namespace experimental } // namespace arm_compute #endif /*ARM_COMPUTE_NERESHAPELAYER_H */ |