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

fix regression

Browse Source 
PiperOrigin-RevId: 255592806
This commit is contained in:
Renjie Liu 2019-06-28 06:48:29 -07:00 committed by TensorFlower Gardener
parent 3438a4c411
commit f0ae8bef6a
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/lite/kernels/internal/optimized/depthwiseconv_uint8.h
View File

@ -2018,8 +2018,8 @@ inline void DepthwiseConvWithRounding(
// Enable for arm64 except for the Nvidia Linux 4 Tegra (L4T) running on // Enable for arm64 except for the Nvidia Linux 4 Tegra (L4T) running on
// Jetson TX-2. This compiler does not support the offsetof() macro. // Jetson TX-2. This compiler does not support the offsetof() macro.
#if defined(__aarch64__) && !defined(GOOGLE_L4T) && defined(__ANDROID__) && \ #if defined(__aarch64__) && !defined(GOOGLE_L4T)
defined(__clang__) #if defined(__ANDROID__) && defined(__clang__)
// Dispatch to dot-product 3x3 kernels when supported. // Dispatch to dot-product 3x3 kernels when supported.
if (cpu_flags.neon_dotprod) { if (cpu_flags.neon_dotprod) {
using optimized_ops::depthwise_conv::DotProduct3x3KernelType; using optimized_ops::depthwise_conv::DotProduct3x3KernelType;
@ -2038,8 +2038,7 @@ inline void DepthwiseConvWithRounding(
} }
} }
#elif defined(__aarch64__) && !defined(GOOGLE_L4T) #endif
// Dispatch to non-dot-product 3x3 kernels when supported. // Dispatch to non-dot-product 3x3 kernels when supported.
const int stride_width = params.stride_width; const int stride_width = params.stride_width;