diff options
author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2018-12-07 11:18:09 +0000 |
---|---|---|
committer | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2018-12-11 14:55:03 +0000 |
commit | 3b0a2654034714c16f5930d2b24936d8be7b18a6 (patch) | |
tree | 2a3ed3703a4f454ef42084363049aba3bf54ebd6 /arm_compute/core/Types.h | |
parent | ff0bccfb4697c591d569db9c2dc223f2e311a7d3 (diff) | |
download | ComputeLibrary-3b0a2654034714c16f5930d2b24936d8be7b18a6.tar.gz |
COMPMID-1775: Implement CLGEMMReshapeRHSMatrixKernel to reshape the RHS matrix of GEMM/GEMMLowp
Change-Id: I77f2bfcc5d170bcc2428a2f27104942c1ec877d7
Reviewed-on: https://review.mlplatform.org/375
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/core/Types.h')
-rw-r--r-- | arm_compute/core/Types.h | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/arm_compute/core/Types.h b/arm_compute/core/Types.h index 55b0ccb30d..6ef9878a95 100644 --- a/arm_compute/core/Types.h +++ b/arm_compute/core/Types.h @@ -1781,6 +1781,16 @@ struct GEMMLHSMatrixInfo bool interleave{ true }; /**< True if the v0 (m0xk0) blocks have to be interleaved in the output row */ }; +/** GEMM RHS (Right Hand Side) matrix information */ +struct GEMMRHSMatrixInfo +{ + unsigned int n0{ 1 }; /**< Number of columns processed by the matrix multiplication */ + unsigned int k0{ 1 }; /**< Number of partial accumulations performed by the matrix multiplication */ + unsigned int h0{ 1 }; /**< Number of horizontal blocks of size (k0xn0) stored on the same output row */ + bool transpose{ true }; /**< True if the (k0xn0) block has to be transposed before been stored */ + bool interleave{ true }; /**< True if the h0 (k0xn0) blocks have to be interleaved in the output row */ +}; + /** GEMM information class. This class stores the necessary information to compute GEMM functions * * This object also contains the information about how matrix A and matrix B have been reshaped |