experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use NEON_OR_PORTABLE with SparseMatrixBatchVectorMultiplyAccumulate

Browse Source 
PiperOrigin-RevId: 253652393
This commit is contained in:
A. Unique TensorFlower 2019-06-17 13:44:17 -07:00 committed by TensorFlower Gardener
parent 524eecf61d
commit 1e1535bbcd
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/lite/kernels/internal/optimized/neon_tensor_utils.h
View File

@ -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,