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/NEGEMMMatrixMultiplyKernel.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/NEGEMMMatrixMultiplyKernel.h')
-rw-r--r-- | arm_compute/core/NEON/kernels/NEGEMMMatrixMultiplyKernel.h | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/arm_compute/core/NEON/kernels/NEGEMMMatrixMultiplyKernel.h b/arm_compute/core/NEON/kernels/NEGEMMMatrixMultiplyKernel.h index aff2671441..9923c31018 100644 --- a/arm_compute/core/NEON/kernels/NEGEMMMatrixMultiplyKernel.h +++ b/arm_compute/core/NEON/kernels/NEGEMMMatrixMultiplyKernel.h @@ -61,6 +61,15 @@ public: * @param[in] alpha Weight of the matrix product */ void configure(const ITensor *input0, const ITensor *input1, ITensor *output, float alpha); + /** Static function to check if given info will lead to a valid configuration of @ref NEGEMMMatrixMultiplyKernel + * + * @param[in] input0 Input tensor containing the Matrix A. Data types supported: QS8/QS16/F16/F32 + * @param[in] input1 Input tensor containing the Matrix B. Data type supported: same as @p input0 + * @param[in] output Output tensor to store the result of matrix multiplication. Data type supported: same as @p input0 + * + * @return a status + */ + static Status validate(const ITensorInfo *input0, const ITensorInfo *input1, const ITensorInfo *output); // Inherited methods overridden: void run(const Window &window, const ThreadInfo &info) override; |