2  * SPDX-FileCopyrightText: Copyright 2024 Arm Limited and/or its affiliates <open-source-office.com>
4  * SPDX-License-Identifier: Apache-2.0
10  * www.apache.org/licenses/LICENSE-2.0
19 /* ----------------------------------------------------------------------
22 …* Description:  Batch matrix multiplication. Does not perform transposes, see header file for deta…
27  * Target :  Arm(R) M-Profile Architecture
29  * -------------------------------------------------------------------- */
44  * Refer to header file for details.
58     if (ctx->buf == NULL)  in arm_batch_matmul_s8()
62     int32_t *vector_sum_buf = (int32_t *)ctx->buf;  in arm_batch_matmul_s8()
64     const int32_t output_batch = output_dims->n;  in arm_batch_matmul_s8()
65     const int32_t output_height = output_dims->h;  in arm_batch_matmul_s8()
66     const int32_t lhs_rows = input_lhs_dims->w;  in arm_batch_matmul_s8()
67     const int32_t rhs_rows = input_rhs_dims->w;  in arm_batch_matmul_s8()
68     const int32_t rhs_cols = input_rhs_dims->c;  in arm_batch_matmul_s8()
70     const int32_t inner_lhs_diff = input_lhs_dims->h >= input_rhs_dims->h ? 0 : lhs_rows * rhs_cols;  in arm_batch_matmul_s8()
71     const int32_t inner_rhs_diff = input_rhs_dims->h >= input_lhs_dims->h ? rhs_rows * rhs_cols : 0;  in arm_batch_matmul_s8()
72     const int32_t outer_lhs_diff = input_lhs_dims->n >= input_rhs_dims->n  in arm_batch_matmul_s8()
74         : -((lhs_rows * rhs_cols) - inner_lhs_diff) * input_lhs_dims->h;  in arm_batch_matmul_s8()
75 …const int32_t outer_rhs_diff = input_rhs_dims->n >= input_lhs_dims->n ? (rhs_rows * rhs_cols) - in…  in arm_batch_matmul_s8()
76 …                                                             : -inner_rhs_diff * input_rhs_dims->h;  in arm_batch_matmul_s8()
88                               bmm_params->fc_params.input_offset,  in arm_batch_matmul_s8()
89                               bmm_params->fc_params.filter_offset,  in arm_batch_matmul_s8()
103                                          bmm_params->fc_params.input_offset,  in arm_batch_matmul_s8()
104                                          bmm_params->fc_params.output_offset,  in arm_batch_matmul_s8()
105                                          quant_params->multiplier,  in arm_batch_matmul_s8()
106                                          quant_params->shift,  in arm_batch_matmul_s8()
109                                          bmm_params->fc_params.activation.min,  in arm_batch_matmul_s8()
110                                          bmm_params->fc_params.activation.max,  in arm_batch_matmul_s8()
112                                          bmm_params->fc_params.filter_offset);  in arm_batch_matmul_s8()
117             input_lhs -= inner_lhs_diff;  in arm_batch_matmul_s8()