diff options
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEMaxUnpoolingLayer.h')
-rw-r--r-- | arm_compute/runtime/NEON/functions/NEMaxUnpoolingLayer.h | 17 |
1 files changed, 10 insertions, 7 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEMaxUnpoolingLayer.h b/arm_compute/runtime/NEON/functions/NEMaxUnpoolingLayer.h index 41ea040457..e00fc4544f 100644 --- a/arm_compute/runtime/NEON/functions/NEMaxUnpoolingLayer.h +++ b/arm_compute/runtime/NEON/functions/NEMaxUnpoolingLayer.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2020-2021 Arm Limited. + * Copyright (c) 2020-2022 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -26,6 +26,7 @@ #include "arm_compute/core/Types.h" #include "arm_compute/runtime/IFunction.h" + #include <memory> namespace arm_compute @@ -33,12 +34,10 @@ namespace arm_compute class ITensor; class ITensorInfo; class NEFill; -class NEMaxUnpoolingLayerKernel; /** Function to perform MaxUnpooling. This function calls the following kernels: * * -# @ref NEFill - * -# @ref NEMaxUnpoolingLayerKernel */ class NEMaxUnpoolingLayer : public IFunction { @@ -88,14 +87,18 @@ public: * * @return a status */ - static Status validate(const ITensorInfo *input, const ITensorInfo *indices, const ITensorInfo *output, const PoolingLayerInfo &pool_info); + static Status validate(const ITensorInfo *input, + const ITensorInfo *indices, + const ITensorInfo *output, + const PoolingLayerInfo &pool_info); // Inherited methods overridden: void run() override; private: - std::unique_ptr<NEFill> _fill_func; - std::unique_ptr<NEMaxUnpoolingLayerKernel> _unpooling_layer_kernel; + std::unique_ptr<NEFill> _fill_func; + struct Impl; + std::unique_ptr<Impl> _impl; }; -} +} // namespace arm_compute #endif /* ARM_COMPUTE_NEMAXUNPOOLINGLAYER_H */ |