diff options
Diffstat (limited to 'src/backends/neon/NeonLayerSupport.cpp')
-rw-r--r-- | src/backends/neon/NeonLayerSupport.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/backends/neon/NeonLayerSupport.cpp b/src/backends/neon/NeonLayerSupport.cpp index 76cdf140d2..2f83c8f82a 100644 --- a/src/backends/neon/NeonLayerSupport.cpp +++ b/src/backends/neon/NeonLayerSupport.cpp @@ -71,6 +71,7 @@ bool IsSupportedForDataTypeNeon(Optional<std::string&> reasonIfUnsupported, floatFuncPtr, floatFuncPtr, uint8FuncPtr, + &FalseFunc<>, std::forward<Params>(params)...); } @@ -212,7 +213,8 @@ bool NeonLayerSupport::IsFloorSupported(const TensorInfo& input, input.GetDataType(), &FalseFuncF16<>, &TrueFunc<>, - &FalseFuncU8<>); + &FalseFuncU8<>, + &FalseFuncI32<>); } bool NeonLayerSupport::IsFullyConnectedSupported(const TensorInfo& input, |