diff options
Diffstat (limited to 'arm_compute')
| -rw-r--r-- | arm_compute/core/Types.h | 5 | ||||
| -rw-r--r-- | arm_compute/runtime/CL/functions/CLDeconvolutionLayer.h | 7 |
2 files changed, 9 insertions, 3 deletions
diff --git a/arm_compute/core/Types.h b/arm_compute/core/Types.h index b0a6475527..d5a4125c88 100644 --- a/arm_compute/core/Types.h +++ b/arm_compute/core/Types.h @@ -150,8 +150,9 @@ enum class DepthwiseConvolutionFunction /** Available DeconvolutionMethod*/ enum class DeconvolutionMethod { - GEMM, /**< Deconvolution using GEMM */ - DIRECT, /**< Direct deconvolution */ + GEMM, /**< Deconvolution using GEMM */ + DIRECT, /**< Direct deconvolution */ + UPSCALE_CONV2D /**< Deconvolution with Upscaling */ }; /** Available FuseBatchNormalizationType*/ diff --git a/arm_compute/runtime/CL/functions/CLDeconvolutionLayer.h b/arm_compute/runtime/CL/functions/CLDeconvolutionLayer.h index 8ad805492d..0c59e2c86d 100644 --- a/arm_compute/runtime/CL/functions/CLDeconvolutionLayer.h +++ b/arm_compute/runtime/CL/functions/CLDeconvolutionLayer.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2021 Arm Limited. + * Copyright (c) 2017-2022 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -44,6 +44,8 @@ public: /** Default constructor */ CLDeconvolutionLayer(std::shared_ptr<IMemoryManager> memory_manager = nullptr); + ~CLDeconvolutionLayer(); + /** Set the input, weights, biases and output tensors. * * Valid data layouts: @@ -105,6 +107,9 @@ public: private: std::shared_ptr<IMemoryManager> _memory_manager; std::unique_ptr<IFunction> _function; + + struct Impl; + std::unique_ptr<Impl> _impl; }; } // namespace arm_compute #endif /* ARM_COMPUTE_CLDECONVOLUTIONLAYER_H */ |