Home
last modified time | relevance | path

Searched defs: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
457 const float batch_scaling_factor = scaling_factors[i / m_rows]; in SseMatrixBatchVectorMultiplyAccumulate() local
H A Dneon_tensor_utils.cc1125 const float batch_scaling_factor = scaling_factors[batch]; in NeonMatrixBatchVectorMultiplyAccumulate() local
1235 const float batch_scaling_factor = scaling_factors[i / m_rows]; in NeonMatrixBatchVectorMultiplyAccumulate() local
1317 const float batch_scaling_factor = scaling_factors[batch]; in NeonMatrixBatchVectorMultiplyAccumulateImpl() local
1471 float batch_scaling_factor = scaling_factors[i / m_rows]; in NeonMatrixBatchVectorMultiplyAccumulate() local
2097 const float batch_scaling_factor = scaling_factors[batch]; in NeonSparseMatrixBatchVectorMultiplyAccumulate() local
H A Dbatch_matmul.h230 const float batch_scaling_factor = scale_ptr2[j]; in BatchMatMul() local
/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
182 const float batch_scaling_factor = scaling_factors[batch]; in PortableMatrixBatchVectorMultiplyAccumulate() local
302 const float batch_scaling_factor = scaling_factors[batch]; in PortableSparseMatrixBatchVectorMultiplyAccumulate() local
H A Dbatch_matmul.h180 const float batch_scaling_factor = scale_ptr2[j]; in BatchMatMul() local