Searched refs:batch_scaling_factor (Results 1 – 5 of 5) sorted by relevance
103 const float batch_scaling_factor = scaling_factors[batch]; in SseMatrixBatchVectorMultiplyAccumulateImpl() local110 per_channel_scale ? per_channel_scale[row] * batch_scaling_factor in SseMatrixBatchVectorMultiplyAccumulateImpl()111 : batch_scaling_factor; in SseMatrixBatchVectorMultiplyAccumulateImpl()254 const float batch_scaling_factor = scaling_factors[i / m_rows]; in SseMatrixBatchVectorMultiplyAccumulate() local256 *result += x * batch_scaling_factor; in SseMatrixBatchVectorMultiplyAccumulate()
230 const float batch_scaling_factor = scale_ptr2[j]; in BatchMatMul() local234 const float32x4_t scaling_factor0 = vdupq_n_f32(batch_scaling_factor); in BatchMatMul()235 const float32x4_t scaling_factor1 = vdupq_n_f32(batch_scaling_factor); in BatchMatMul()263 out_ptr[idx] += batch_scaling_factor * accum_scratch[idx]; in BatchMatMul()
1099 const float batch_scaling_factor = scaling_factors[batch]; in NeonMatrixBatchVectorMultiplyAccumulate() local1165 *result += dotprod * batch_scaling_factor; in NeonMatrixBatchVectorMultiplyAccumulate()1209 const float batch_scaling_factor = scaling_factors[i / m_rows]; in NeonMatrixBatchVectorMultiplyAccumulate() local1211 *result += x * batch_scaling_factor; in NeonMatrixBatchVectorMultiplyAccumulate()1291 const float batch_scaling_factor = scaling_factors[batch]; in NeonMatrixBatchVectorMultiplyAccumulateImpl() local1300 float scale = batch_scaling_factor; in NeonMatrixBatchVectorMultiplyAccumulateImpl()1448 float batch_scaling_factor = scaling_factors[i / m_rows]; in NeonMatrixBatchVectorMultiplyAccumulate() local1450 batch_scaling_factor *= per_channel_scale[i % m_rows]; in NeonMatrixBatchVectorMultiplyAccumulate()1455 *result += dotprod * batch_scaling_factor; in NeonMatrixBatchVectorMultiplyAccumulate()1997 const float batch_scaling_factor = scaling_factors[batch]; in NeonSparseMatrixBatchVectorMultiplyAccumulate() local[all …]
143 const float batch_scaling_factor = scaling_factors[batch]; in PortableMatrixBatchVectorMultiplyAccumulate() local157 *result += dotprod * batch_scaling_factor; in PortableMatrixBatchVectorMultiplyAccumulate()182 const float batch_scaling_factor = scaling_factors[batch]; in PortableMatrixBatchVectorMultiplyAccumulate() local187 float scale = batch_scaling_factor; in PortableMatrixBatchVectorMultiplyAccumulate()267 const float batch_scaling_factor = scaling_factors[batch]; in PortableSparseMatrixBatchVectorMultiplyAccumulate() local287 result[batch * m_rows + row] += dotprod * batch_scaling_factor; in PortableSparseMatrixBatchVectorMultiplyAccumulate()
179 const float batch_scaling_factor = scale_ptr2[j]; in BatchMatMul() local190 out_ptr[idx] += batch_scaling_factor * total; in BatchMatMul()