diff options
author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2017-08-08 08:38:09 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-09-17 14:18:34 +0100 |
commit | 559d77179bfe8daaeb1f6754ee2287bd12a6aa04 (patch) | |
tree | f9a39dd27b1721f1f6a2794dcfde9a34cba03480 /arm_compute/core/Types.h | |
parent | 6203153b25fda2c4b8d94fe71337d1145a36c132 (diff) | |
download | ComputeLibrary-559d77179bfe8daaeb1f6754ee2287bd12a6aa04.tar.gz |
COMPMID-417 - Fixed auto-config in NEConvolutionLayer and in CLConvolutionLayer
Change-Id: Ibfd772200348b326738bb3b8357f0abbb7a583d7
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/82943
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-by: Moritz Pflanzer <moritz.pflanzer@arm.com>
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Diffstat (limited to 'arm_compute/core/Types.h')
-rw-r--r-- | arm_compute/core/Types.h | 18 |
1 files changed, 14 insertions, 4 deletions
diff --git a/arm_compute/core/Types.h b/arm_compute/core/Types.h index 3092976149..1d04f35359 100644 --- a/arm_compute/core/Types.h +++ b/arm_compute/core/Types.h @@ -607,13 +607,13 @@ private: float _kappa; }; -/** Convolution Layer Weights Information class */ +/** Convolution Layer Weights Information class. This class stores the necessary information to compute convolution layer when the weights are already reshaped */ class WeightsInfo { public: /** Default constructor */ WeightsInfo() - : _are_reshaped(false), _kernel_width(0), _kernel_height(0) + : _are_reshaped(false), _kernel_width(0), _kernel_height(0), _num_kernels(0) { } /** Constructor @@ -621,9 +621,10 @@ public: * @param[in] are_reshaped True if the weights have been reshaped * @param[in] kernel_width Kernel width. * @param[in] kernel_height Kernel height. + * @param[in] num_kernels Number of convolution kernels. */ - WeightsInfo(bool are_reshaped, unsigned int kernel_width, unsigned int kernel_height) - : _are_reshaped(are_reshaped), _kernel_width(kernel_width), _kernel_height(kernel_height) + WeightsInfo(bool are_reshaped, unsigned int kernel_width, unsigned int kernel_height, unsigned int num_kernels) + : _are_reshaped(are_reshaped), _kernel_width(kernel_width), _kernel_height(kernel_height), _num_kernels(num_kernels) { } /** Flag which specifies if the weights tensor has been reshaped. @@ -634,6 +635,14 @@ public: { return _are_reshaped; }; + /** Return the number of convolution kernels + * + * @return The number of convolution kernels + */ + unsigned int num_kernels() const + { + return _num_kernels; + }; /** Return the width and height of the kernel * * @return The width and height of the kernel @@ -647,6 +656,7 @@ private: const bool _are_reshaped; const unsigned int _kernel_width; const unsigned int _kernel_height; + const unsigned int _num_kernels; }; /** IO formatting information class*/ |