Searched refs:batch_scaling_factor (Results 1 – 5 of 5) sorted by relevance
206 const float batch_scaling_factor = scaling_factors[batch]; in Avx2MatrixBatchVectorMultiplyAccumulateImpl() local213 per_channel_scale ? per_channel_scale[row] * batch_scaling_factor in Avx2MatrixBatchVectorMultiplyAccumulateImpl()214 : batch_scaling_factor; in Avx2MatrixBatchVectorMultiplyAccumulateImpl()304 const float batch_scaling_factor = scaling_factors[batch]; in SseMatrixBatchVectorMultiplyAccumulateImpl()311 per_channel_scale ? per_channel_scale[row] * batch_scaling_factor in SseMatrixBatchVectorMultiplyAccumulateImpl()312 : batch_scaling_factor; in SseMatrixBatchVectorMultiplyAccumulateImpl()457 const float batch_scaling_factor = scaling_factors[i / m_rows]; in SseMatrixBatchVectorMultiplyAccumulate() local459 *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()
1125 const float batch_scaling_factor = scaling_factors[batch]; in NeonMatrixBatchVectorMultiplyAccumulate() local1191 *result += dotprod * batch_scaling_factor; in NeonMatrixBatchVectorMultiplyAccumulate()1235 const float batch_scaling_factor = scaling_factors[i / m_rows]; in NeonMatrixBatchVectorMultiplyAccumulate() local1237 *result += x * batch_scaling_factor; in NeonMatrixBatchVectorMultiplyAccumulate()1317 const float batch_scaling_factor = scaling_factors[batch]; in NeonMatrixBatchVectorMultiplyAccumulateImpl() local1326 float scale = batch_scaling_factor; in NeonMatrixBatchVectorMultiplyAccumulateImpl()1471 float batch_scaling_factor = scaling_factors[i / m_rows]; in NeonMatrixBatchVectorMultiplyAccumulate() local1473 batch_scaling_factor *= per_channel_scale[i % m_rows]; in NeonMatrixBatchVectorMultiplyAccumulate()1478 *result += dotprod * batch_scaling_factor; in NeonMatrixBatchVectorMultiplyAccumulate()2097 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()302 const float batch_scaling_factor = scaling_factors[batch]; in PortableSparseMatrixBatchVectorMultiplyAccumulate() local322 result[batch * m_rows + row] += dotprod * batch_scaling_factor; in PortableSparseMatrixBatchVectorMultiplyAccumulate()
180 const float batch_scaling_factor = scale_ptr2[j]; in BatchMatMul() local191 out_ptr[idx] += batch_scaling_factor * total; in BatchMatMul()