From 6b3ae06ec6785bbf20d729bdd3466d88dff97750 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Wed, 19 Jun 2019 16:47:20 -0700 Subject: [PATCH] Don't define Neon* functions when USE_NEON define is missing. PiperOrigin-RevId: 254098359 --- .../lite/kernels/internal/optimized/neon_tensor_utils_impl.h | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tensorflow/lite/kernels/internal/optimized/neon_tensor_utils_impl.h b/tensorflow/lite/kernels/internal/optimized/neon_tensor_utils_impl.h index 8a173e4622a..40281416338 100644 --- a/tensorflow/lite/kernels/internal/optimized/neon_tensor_utils_impl.h +++ b/tensorflow/lite/kernels/internal/optimized/neon_tensor_utils_impl.h @@ -27,6 +27,8 @@ limitations under the License. namespace tflite { namespace tensor_utils { +#ifdef USE_NEON + // Multiply a matrix by a batch vector, and store results in a batch-size // vector. 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, int output_size, int reduction_size); +#endif // USE_NEON + } // namespace tensor_utils } // namespace tflite