Home
last modified time | relevance | path

Searched refs:batch_scaling_factor (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dsse_tensor_utils.cc206 const float batch_scaling_factor = scaling_factors[batch]; in Avx2MatrixBatchVectorMultiplyAccumulateImpl() local
213 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() local
459 *result += x * batch_scaling_factor; in SseMatrixBatchVectorMultiplyAccumulate()
H A Dbatch_matmul.h230 const float batch_scaling_factor = scale_ptr2[j]; in BatchMatMul() local
234 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()
H A Dneon_tensor_utils.cc1125 const float batch_scaling_factor = scaling_factors[batch]; in NeonMatrixBatchVectorMultiplyAccumulate() local
1191 *result += dotprod * batch_scaling_factor; in NeonMatrixBatchVectorMultiplyAccumulate()
1235 const float batch_scaling_factor = scaling_factors[i / m_rows]; in NeonMatrixBatchVectorMultiplyAccumulate() local
1237 *result += x * batch_scaling_factor; in NeonMatrixBatchVectorMultiplyAccumulate()
1317 const float batch_scaling_factor = scaling_factors[batch]; in NeonMatrixBatchVectorMultiplyAccumulateImpl() local
1326 float scale = batch_scaling_factor; in NeonMatrixBatchVectorMultiplyAccumulateImpl()
1471 float batch_scaling_factor = scaling_factors[i / m_rows]; in NeonMatrixBatchVectorMultiplyAccumulate() local
1473 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 …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dportable_tensor_utils.cc143 const float batch_scaling_factor = scaling_factors[batch]; in PortableMatrixBatchVectorMultiplyAccumulate() local
157 *result += dotprod * batch_scaling_factor; in PortableMatrixBatchVectorMultiplyAccumulate()
182 const float batch_scaling_factor = scaling_factors[batch]; in PortableMatrixBatchVectorMultiplyAccumulate() local
187 float scale = batch_scaling_factor; in PortableMatrixBatchVectorMultiplyAccumulate()
302 const float batch_scaling_factor = scaling_factors[batch]; in PortableSparseMatrixBatchVectorMultiplyAccumulate() local
322 result[batch * m_rows + row] += dotprod * batch_scaling_factor; in PortableSparseMatrixBatchVectorMultiplyAccumulate()
H A Dbatch_matmul.h180 const float batch_scaling_factor = scale_ptr2[j]; in BatchMatMul() local
191 out_ptr[idx] += batch_scaling_factor * total; in BatchMatMul()