diff options
Diffstat (limited to 'runtimes/neurun/src/kernel/cpu/ReshapeLayer.h')
-rw-r--r-- | runtimes/neurun/src/kernel/cpu/ReshapeLayer.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/runtimes/neurun/src/kernel/cpu/ReshapeLayer.h b/runtimes/neurun/src/kernel/cpu/ReshapeLayer.h index 395cc1d7f..51d0bacee 100644 --- a/runtimes/neurun/src/kernel/cpu/ReshapeLayer.h +++ b/runtimes/neurun/src/kernel/cpu/ReshapeLayer.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 ReshapeLayer : public ::arm_compute::IFunction +class ReshapeLayer : public ::neurun::exec::IFunction { public: ReshapeLayer(); |