diff --git a/tensorflow/lite/kernels/internal/optimized/neon_tensor_utils.h b/tensorflow/lite/kernels/internal/optimized/neon_tensor_utils.h index a74d544a79d..c4f13a17153 100644 --- a/tensorflow/lite/kernels/internal/optimized/neon_tensor_utils.h +++ b/tensorflow/lite/kernels/internal/optimized/neon_tensor_utils.h @@ -45,8 +45,8 @@ void SparseMatrixBatchVectorMultiplyAccumulate( const float* __restrict__ matrix, const uint8_t* __restrict__ ledger, int m_rows, int m_cols, const float* __restrict__ vector, int n_batch, float* __restrict__ result, int result_stride) { - NeonSparseMatrixBatchVectorMultiplyAccumulate( - matrix, ledger, m_rows, m_cols, vector, n_batch, result, result_stride); + NEON_OR_PORTABLE(SparseMatrixBatchVectorMultiplyAccumulate, matrix, ledger, + m_rows, m_cols, vector, n_batch, result, result_stride); } void SparseMatrixBatchVectorMultiplyAccumulate( @@ -54,9 +54,9 @@ void SparseMatrixBatchVectorMultiplyAccumulate( const int m_cols, const int8_t* __restrict__ vectors, const float* scaling_factors, int n_batch, float* __restrict__ result, int result_stride) { - NeonSparseMatrixBatchVectorMultiplyAccumulate(matrix, ledger, m_rows, m_cols, - vectors, scaling_factors, - n_batch, result, result_stride); + NEON_OR_PORTABLE(SparseMatrixBatchVectorMultiplyAccumulate, matrix, ledger, + m_rows, m_cols, vectors, scaling_factors, n_batch, result, + result_stride); } void VectorVectorCwiseProduct(const float* vector1, const float* vector2,