diff options
Diffstat (limited to 'runtimes/neurun/src/kernel/cpu/MaxPoolLayer.h')
-rw-r--r-- | runtimes/neurun/src/kernel/cpu/MaxPoolLayer.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/runtimes/neurun/src/kernel/cpu/MaxPoolLayer.h b/runtimes/neurun/src/kernel/cpu/MaxPoolLayer.h index b42efb9f6..2b185550b 100644 --- a/runtimes/neurun/src/kernel/cpu/MaxPoolLayer.h +++ b/runtimes/neurun/src/kernel/cpu/MaxPoolLayer.h @@ -19,7 +19,7 @@ #include <NeuralNetworks.h> -#include <arm_compute/runtime/IFunction.h> +#include "exec/interface/IFunction.h" #include "kernel/cpu/OperationUtils.h" @@ -30,7 +30,7 @@ namespace kernel namespace cpu { -class MaxPoolLayer : public ::arm_compute::IFunction +class MaxPoolLayer : public ::neurun::exec::IFunction { public: MaxPoolLayer(); |