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

Don't define Neon* functions when USE_NEON define is missing.

Browse Source 
PiperOrigin-RevId: 254098359
This commit is contained in:
A. Unique TensorFlower 2019-06-19 16:47:20 -07:00 committed by TensorFlower Gardener
parent 835bcfba8f
commit 6b3ae06ec6
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/lite/kernels/internal/optimized/neon_tensor_utils_impl.h
View File

@ -27,6 +27,8 @@ limitations under the License.
namespace tflite { namespace tflite {
namespace tensor_utils { namespace tensor_utils {
#ifdef USE_NEON
// Multiply a matrix by a batch vector, and store results in a batch-size // Multiply a matrix by a batch vector, and store results in a batch-size
// vector. // vector.
void NeonMatrixBatchVectorMultiplyAccumulate(const float* matrix, int m_rows, void NeonMatrixBatchVectorMultiplyAccumulate(const float* matrix, int m_rows,
@ -119,6 +121,8 @@ void NeonVectorShiftLeft(float* vector, int v_size, float shift_value);
void NeonReductionSumVector(const float* input_vector, float* output_vector, void NeonReductionSumVector(const float* input_vector, float* output_vector,
int output_size, int reduction_size); int output_size, int reduction_size);
#endif // USE_NEON
} // namespace tensor_utils } // namespace tensor_utils
} // namespace tflite } // namespace tflite