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

Fix build of neon_tensor_utils for android_x86

Browse Source 
PiperOrigin-RevId: 291833076
Change-Id: I93398a8f46101708d1d010573f1216473fbdee16
This commit is contained in:
T.J. Alumbaugh 2020-01-27 17:07:47 -08:00 committed by TensorFlower Gardener
parent 2cf7afb04a
commit 48a0b665b8
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/lite/kernels/internal/reference/portable_tensor_utils_impl.h
View File

@ -67,6 +67,12 @@ void PortableMatrixBatchVectorMultiplyAccumulate(
const int8_t* __restrict__ vectors, const float* scaling_factors, const int8_t* __restrict__ vectors, const float* scaling_factors,
int n_batch, float* __restrict__ result, int result_stride); int n_batch, float* __restrict__ result, int result_stride);
void PortableMatrixBatchVectorMultiplyAccumulate(
const int8_t* __restrict__ matrix, const int m_rows, const int m_cols,
const int8_t* __restrict__ vector, const float* scaling_factors,
int n_batch, int32_t* scratch, float* __restrict__ result,
int result_stride, CpuBackendContext* context);
void PortableMatrixBatchVectorMultiplyAccumulate( void PortableMatrixBatchVectorMultiplyAccumulate(
const int8_t* __restrict__ matrix, const int m_rows, const int m_cols, const int8_t* __restrict__ matrix, const int m_rows, const int m_cols,
const int8_t* __restrict__ vectors, const float* scaling_factors, const int8_t* __restrict__ vectors, const float* scaling_factors,