Searched refs:inner_rhs_diff (Results 1 – 2 of 2) sorted by relevance
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() local68 …_t outer_rhs_diff = input_rhs_dims->n >= input_lhs_dims->n ? (rhs_rows * rhs_cols) - inner_rhs_diff in arm_batch_matmul_s16()69 … : -inner_rhs_diff * input_rhs_dims->h; in arm_batch_matmul_s16()94 input_rhs += inner_rhs_diff; in arm_batch_matmul_s16()
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() local75 …_t outer_rhs_diff = input_rhs_dims->n >= input_lhs_dims->n ? (rhs_rows * rhs_cols) - inner_rhs_diff in arm_batch_matmul_s8()76 … : -inner_rhs_diff * input_rhs_dims->h; in arm_batch_matmul_s8()118 input_rhs += inner_rhs_diff; in arm_batch_matmul_s8()