diff options
author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2018-12-06 17:13:09 +0000 |
---|---|---|
committer | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2018-12-10 15:58:54 +0000 |
commit | 5ba5e0938e68d4f90f5545a81066d56f022b376a (patch) | |
tree | d828f8b3fd52e6d5b8f732a7ec41f832f0b921d8 /arm_compute/core/Types.h | |
parent | 1d7cbb99d2a34abd15f3b6c2e017115736cd90cc (diff) | |
download | ComputeLibrary-5ba5e0938e68d4f90f5545a81066d56f022b376a.tar.gz |
COMPMID-1774: Implement CLGEMMReshapeLHSMatrixKernel to reshape the LHS matrix of GEMM/GEMMLowp
Change-Id: I8c5fd4c8bcdffda1522c83158981ed92baa045f4
Reviewed-on: https://review.mlplatform.org/364
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 75b38c5cb8..55b0ccb30d 100644 --- a/arm_compute/core/Types.h +++ b/arm_compute/core/Types.h @@ -1771,6 +1771,16 @@ struct GEMMLowpOutputStageInfo int gemmlowp_max_bound{ 0 }; /**< GEMMLowp max value used to saturate down the output result before converting back to QASYMM8 */ }; +/** GEMM LHS (Left Hand Side) matrix information */ +struct GEMMLHSMatrixInfo +{ + unsigned int m0{ 1 }; /**< Number of rows processed by the matrix multiplication */ + unsigned int k0{ 1 }; /**< Number of partial accumulations performed by the matrix multiplication */ + unsigned int v0{ 1 }; /**< Number of vertical blocks of size (m0xk0) stored on the same output row */ + bool transpose{ true }; /**< True if the (m0xk0) block has to be transposed before been stored */ + bool interleave{ true }; /**< True if the v0 (m0xk0) 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 |