diff options
Diffstat (limited to 'src/cpu')
-rw-r--r-- | src/cpu/kernels/cast/generic/neon/bfloat16.cpp | 1 | ||||
-rw-r--r-- | src/cpu/kernels/cast/generic/neon/fp16.cpp | 1 | ||||
-rw-r--r-- | src/cpu/operators/CpuConv2d.cpp | 1 |
3 files changed, 3 insertions, 0 deletions
diff --git a/src/cpu/kernels/cast/generic/neon/bfloat16.cpp b/src/cpu/kernels/cast/generic/neon/bfloat16.cpp index 91c15be279..d8e2756192 100644 --- a/src/cpu/kernels/cast/generic/neon/bfloat16.cpp +++ b/src/cpu/kernels/cast/generic/neon/bfloat16.cpp @@ -23,6 +23,7 @@ */ #if defined(ARM_COMPUTE_ENABLE_BF16) +#include "arm_compute/core/CPP/CPPTypes.h" #include "arm_compute/core/TensorInfo.h" #include "src/core/NEON/wrapper/wrapper.h" #include "src/cpu/kernels/CpuCastKernel.h" diff --git a/src/cpu/kernels/cast/generic/neon/fp16.cpp b/src/cpu/kernels/cast/generic/neon/fp16.cpp index 385ca1898d..6cd0c8500b 100644 --- a/src/cpu/kernels/cast/generic/neon/fp16.cpp +++ b/src/cpu/kernels/cast/generic/neon/fp16.cpp @@ -23,6 +23,7 @@ */ #if defined(__ARM_FEATURE_FP16_VECTOR_ARITHMETIC) && defined(ENABLE_FP16_KERNELS) +#include "arm_compute/core/CPP/CPPTypes.h" #include "arm_compute/core/TensorInfo.h" #include "src/cpu/kernels/CpuCastKernel.h" #include "src/cpu/kernels/cast/list.h" diff --git a/src/cpu/operators/CpuConv2d.cpp b/src/cpu/operators/CpuConv2d.cpp index 447b740989..16ac16b3ba 100644 --- a/src/cpu/operators/CpuConv2d.cpp +++ b/src/cpu/operators/CpuConv2d.cpp @@ -215,6 +215,7 @@ ConvolutionMethod CpuConv2d::get_convolution_method(const ITensorInfo *input, co } } #endif // __ARM_FEATURE_FP16_VECTOR_ARITHMETIC + // For 1x1 convolutions run the default GEMM if(weights->dimension(idx_w) == 1 && weights->dimension(idx_h) == 1) { |