diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-10-14 19:03:09 +0100 |
|---|---|---|
| committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-10-23 12:08:12 +0000 |
| commit | 48b3ef89de5f21a0169d8416e3d54081f82c7bf8 (patch) | |
| tree | f857d733ccf446c704823dc7ac796a96eb55095e /arm_compute/core/NEON/kernels/assembly/Helpers.h | |
| parent | 1dce3101ef8d77c8cf0af7dfd4af6595a0136b91 (diff) | |
| download | ComputeLibrary-48b3ef89de5f21a0169d8416e3d54081f82c7bf8.tar.gz | |
COMPMID-2577: Fuse bias addition and activation in gemm assembly kernels
Change-Id: I7f52112d2d05b1ea3d3f3d4b19b8eafab05d6c44
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-on: https://review.mlplatform.org/c/2141
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Pablo Marquez <pablo.tello@arm.com>
Diffstat (limited to 'arm_compute/core/NEON/kernels/assembly/Helpers.h')
| -rw-r--r-- | arm_compute/core/NEON/kernels/assembly/Helpers.h | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/arm_compute/core/NEON/kernels/assembly/Helpers.h b/arm_compute/core/NEON/kernels/assembly/Helpers.h index e2a46e96a3..092ce400d1 100644 --- a/arm_compute/core/NEON/kernels/assembly/Helpers.h +++ b/arm_compute/core/NEON/kernels/assembly/Helpers.h @@ -47,8 +47,7 @@ struct BlockSizes * @param[in] ci CPU information. * @param[in] num_threads Maximum number of threads that might be used for the calculations. * @param[in] p M, N, K sizes. - * @param[in] alpha Alpha value. - * @param[in] beta Beta value. + * @param[in] activation Activation struct * @param[in] pretranspose_hint Is B also pretransposed ? * * @return Kernel description that the assembly heuristics picked for the given configuration @@ -57,8 +56,7 @@ arm_gemm::KernelDescription get_gemm_info(DataType in const CPUInfo &ci, const unsigned int num_threads, const INEGEMMWrapperKernel::Params &p, - float alpha, - float beta, + arm_gemm::Activation activation, bool pretranspose_hint); /** Calculate the recommended block sizes to use based on the CPU cache sizes and the strategy which will be used |