Lines Matching refs:rhs_offset
134 int rhs_offset, int result_offset, int result_mult_int, in Gemm()
145 const OffsetRowDup rhs_offset_vector(rhs_offset, cols); in Gemm()
174 int rhs_offset, int result_offset, int result_mult_int, in Gemm()
186 const OffsetRowDup rhs_offset_vector(rhs_offset, cols); in Gemm()
211 int rhs_offset, int result_offset, int result_mult_int, in Gemm()
216 rhs_offset, result_offset, result_mult_int, in Gemm()
245 int rhs_offset, int result_offset, int result_mult_int, in Gemm()
254 rhs_offset, rhs.stride(), result->data(), result_offset, in Gemm()
271 int rhs_offset, int result_offset, int result_mult_int, in Gemm()
276 lhs.stride(), rhs.data(), rhs_offset, rhs.stride(), in Gemm()
403 int rhs_offset, int result_offset, int result_mult_int, in test_gemm_impl() argument
413 &result->map(), lhs_offset, rhs_offset, result_offset, in test_gemm_impl()
431 &ref_result.map(), lhs_offset, rhs_offset, result_offset, result_mult_int, in test_gemm_impl()
457 rhs_offset, result_offset, result_mult_int, in test_gemm_impl()
472 lhs_offset, rhs_offset, result_offset, result_mult_int, result_shift); in test_gemm_impl()
496 int rhs_offset, int result_offset, int result_mult_int) { in test_gemm() argument
497 test_gemm_impl<GemmWrapper>(context, lhs, rhs, result, lhs_offset, rhs_offset, in test_gemm()
760 const OffsetRowDup rhs_offset(b_offset, rhs.cols()); in TestWithSmallDataPerChannelQuantization() local
784 &gemm_context, lhs, rhs, &result, lhs_offset, rhs_offset, in TestWithSmallDataPerChannelQuantization()
911 const OffsetRowDup rhs_offset(b_offset, rhs.cols()); in TestWithLargeDataPerChannelQuantization() local
981 &gemm_context, lhs, rhs, &result, lhs_offset, rhs_offset, in TestWithLargeDataPerChannelQuantization()
1053 const OffsetRowDup rhs_offset(b_offset, rhs.cols()); in TestMultithreadedPerChannelQuantization() local
1094 &gemm_context, lhs, rhs, &result, lhs_offset, rhs_offset, in TestMultithreadedPerChannelQuantization()
1209 const int rhs_offset = -34; in TestOutputStages() local
1216 rhs_offset, empty_pipeline); in TestOutputStages()
1225 static_cast<std::int32_t>(rhs(d, c)) + rhs_offset; in TestOutputStages()
1242 lhs_offset, rhs_offset, quantize_down_pipeline); in TestOutputStages()
1269 &result_quantized_down_saturated_uint8, lhs_offset, rhs_offset, in TestOutputStages()
1289 &result_quantized_down_saturated_int16, lhs_offset, rhs_offset, in TestOutputStages()
1309 &result_quantized_down_truncated_uint8, lhs_offset, rhs_offset, in TestOutputStages()
1336 lhs_offset, rhs_offset, row_bias_addition_pipeline); in TestOutputStages()
1357 lhs_offset, rhs_offset, col_bias_addition_pipeline); in TestOutputStages()
1385 rhs_offset, clamp_pipeline); in TestOutputStages()
1405 rhs_offset, tanh_pipeline); in TestOutputStages()
1425 lhs_offset, rhs_offset, bias_clamp_pipeline); in TestOutputStages()
1445 &result_biased_clamped_quantized_casted, lhs_offset, rhs_offset, in TestOutputStages()
1488 &result_quantized_down_by_fixedpoint_int32, lhs_offset, rhs_offset, in TestOutputStages()
1522 &result_scaled_by_fixedpoint_and_exponent_int32, lhs_offset, rhs_offset, in TestOutputStages()
1554 rhs_offset, quantize_down_by_fixedpoint_and_saturating_cast_pipeline); in TestOutputStages()