diff options
author | Giorgio Arena <giorgio.arena@arm.com> | 2017-11-30 15:08:38 +0000 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:42:33 +0000 |
commit | 7c23ad01c028f73aef0b439fc5d5d14e92e5f4e2 (patch) | |
tree | b6649124c552bafe3b9bfb8f62e6536f27215173 /arm_compute/core/NEON/kernels/NEWeightsReshapeKernel.h | |
parent | 926d5e17d073cf7f0999d4ab4ce4f79f5391abb8 (diff) | |
download | ComputeLibrary-7c23ad01c028f73aef0b439fc5d5d14e92e5f4e2.tar.gz |
COMPMID-617 Add validation to NEConvolutionLayer
Change-Id: I796a13e6ea672e274aaa8234ee0689828fec7292
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/111348
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Ioan-Cristian Szabo <ioan-cristian.szabo@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'arm_compute/core/NEON/kernels/NEWeightsReshapeKernel.h')
-rw-r--r-- | arm_compute/core/NEON/kernels/NEWeightsReshapeKernel.h | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/arm_compute/core/NEON/kernels/NEWeightsReshapeKernel.h b/arm_compute/core/NEON/kernels/NEWeightsReshapeKernel.h index 5b37d6cc62..84ec736d03 100644 --- a/arm_compute/core/NEON/kernels/NEWeightsReshapeKernel.h +++ b/arm_compute/core/NEON/kernels/NEWeightsReshapeKernel.h @@ -77,6 +77,17 @@ public: * @param[out] output The output tensor. Data types supported: Same as @p input */ void configure(const ITensor *input, const ITensor *bias, ITensor *output); + /** Static function to check if given info will lead to a valid configuration of @ref NEWeightsReshapeKernel + * + * @param[in] input The input tensor to convert. Weights are 4D tensor with dimensions [kernel_x, kernel_y, IFM, OFM] if shared, + * and 5D tensor with dimensions [kernel_x, kernel_y, IFM, OFM, num_patches] if unshared. Data types supported: QS8/QS16/F16/F32 + * @param[in] biases The shared biases tensor to append. Bias is 1D tensor with dimensions [OFM] if shared and 2D tensor with + * dimensions [OFM, num_patches] if unshared. Data types supported: Same as @p input + * @param[in] output The output tensor. Should be a 2D Tensor. Data types supported: Same as @p input + * + * @return a status + */ + static Status validate(const ITensorInfo *input, const ITensorInfo *biases, const ITensorInfo *output); // Inherited methods overridden: void run(const Window &window, const ThreadInfo &info) override; |