Lines Matching +full:- +full:- +full:header
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.
57 const int32_t output_batch = output_dims->n; in arm_batch_matmul_s16()
58 const int32_t output_height = output_dims->h; in arm_batch_matmul_s16()
59 const int32_t lhs_rows = input_lhs_dims->w; in arm_batch_matmul_s16()
60 const int32_t rhs_rows = input_rhs_dims->w; in arm_batch_matmul_s16()
61 const int32_t rhs_cols = input_rhs_dims->c; in arm_batch_matmul_s16()
63 const int32_t inner_lhs_diff = input_lhs_dims->h >= input_rhs_dims->h ? 0 : lhs_rows * rhs_cols; in arm_batch_matmul_s16()
64 const int32_t inner_rhs_diff = input_rhs_dims->h >= input_lhs_dims->h ? rhs_rows * rhs_cols : 0; in arm_batch_matmul_s16()
65 const int32_t outer_lhs_diff = input_lhs_dims->n >= input_rhs_dims->n in arm_batch_matmul_s16()
67 : -((lhs_rows * rhs_cols) - inner_lhs_diff) * input_lhs_dims->h; in arm_batch_matmul_s16()
68 …const int32_t outer_rhs_diff = input_rhs_dims->n >= input_lhs_dims->n ? (rhs_rows * rhs_cols) - in… in arm_batch_matmul_s16()
69 … : -inner_rhs_diff * input_rhs_dims->h; in arm_batch_matmul_s16()
71 const int32_t reduced_multiplier = REDUCE_MULTIPLIER(quant_params->multiplier); in arm_batch_matmul_s16()
85 quant_params->shift, in arm_batch_matmul_s16()
88 bmm_params->fc_params.activation.min, in arm_batch_matmul_s16()
89 bmm_params->fc_params.activation.max); in arm_batch_matmul_s16()
93 input_lhs -= inner_lhs_diff; in arm_batch_matmul_s16()