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

Use sse_tensor_utils when SSSE 3 is available instead of just SSE 4.1.

Browse Source 
PiperOrigin-RevId: 293209198
Change-Id: I88c70743f6deaff5a6fdb2b4b4a7c5118b3edf63
This commit is contained in:
Robert David 2020-02-04 12:58:05 -08:00 committed by TensorFlower Gardener
parent bd23bbf19c
commit 9bdcb357b5
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/lite/kernels/internal/tensor_utils.cc
View File

@ -16,10 +16,10 @@ limitations under the License.
#include "tensorflow/lite/kernels/internal/optimized/neon_check.h" #include "tensorflow/lite/kernels/internal/optimized/neon_check.h"
#if defined(__SSE4_1__) #if defined(__SSSE3__)
#include "tensorflow/lite/kernels/internal/optimized/sse_tensor_utils.h" #include "tensorflow/lite/kernels/internal/optimized/sse_tensor_utils.h"
#elif defined(USE_NEON) #elif defined(USE_NEON)
#include "tensorflow/lite/kernels/internal/optimized/neon_tensor_utils.h" #include "tensorflow/lite/kernels/internal/optimized/neon_tensor_utils.h"
#else #else
#include "tensorflow/lite/kernels/internal/reference/portable_tensor_utils.h" #include "tensorflow/lite/kernels/internal/reference/portable_tensor_utils.h"
#endif // __SSE4_1__ or USE_NEON #endif // __SSSE3__ or USE_NEON