aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core/CPP
diff options
context:
space:
mode:
authorMichele Di Giorgio <michele.digiorgio@arm.com>2018-09-13 16:22:01 +0100
committerAnthony Barbier <anthony.barbier@arm.com>2018-11-02 16:55:45 +0000
commited5a492ba791d8c8b3334749d4ae946b8f11d13d (patch)
tree89c8cd6f705dc88a21c61668164aad079800aff7 /arm_compute/core/CPP
parent7e9391bb14d219cda310bff355669b5964b1f576 (diff)
downloadComputeLibrary-ed5a492ba791d8c8b3334749d4ae946b8f11d13d.tar.gz
COMPMID-1586: Add support for NHWC CLDeconvolutionLayer
COMPMID-1651: Fix QASYMM8 CLDeconvolutionLayer This patch also extends the range of values used for testing Convolution and Deconvolution to cover quantized [-1.0f, 1.0f]. Change-Id: I8b280669db67bb3ec25bf5d411c8f5954f5b0dab Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/149869 Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com> Tested-by: bsgcomp <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/core/CPP')
-rw-r--r--arm_compute/core/CPP/kernels/CPPFlipWeightsKernel.h8
1 files changed, 3 insertions, 5 deletions
diff --git a/arm_compute/core/CPP/kernels/CPPFlipWeightsKernel.h b/arm_compute/core/CPP/kernels/CPPFlipWeightsKernel.h
index 801934159d..04567ed959 100644
--- a/arm_compute/core/CPP/kernels/CPPFlipWeightsKernel.h
+++ b/arm_compute/core/CPP/kernels/CPPFlipWeightsKernel.h
@@ -53,7 +53,7 @@ public:
/** Set the input and output of the kernel.
*
- * @param[in] input The input tensor to flip. Data types supported: QASYMM8/F16/F32
+ * @param[in] input The input tensor to flip. Data types supported: QASYMM8/F16/F32. Data layouts supported: NCHW/NHWC.
* @param[out] output The output tensor. Data types supported: Same as @p input
*/
void configure(const ITensor *input, ITensor *output);
@@ -64,17 +64,15 @@ public:
/** Function to perform flipping.
*
* @param[in] window_input Input region on which to execute the kernel.
- * @param[in] window Output region on which to execute the kernel.
*/
template <typename T>
- void flip_weights(const Window &window_input, const Window &window);
+ void flip_weights(const Window &window_input);
/** Common signature for all the specialised Flip functions
*
* @param[in] window_input Input region on which to execute the kernel.
- * @param[in] window Output region on which to execute the kernel.
*/
- using FlipWeightsFunction = void (CPPFlipWeightsKernel::*)(const Window &window_input, const Window &window);
+ using FlipWeightsFunction = void (CPPFlipWeightsKernel::*)(const Window &window_input);
private:
const ITensor *_input;