summaryrefslogtreecommitdiff
path: root/runtimes/neurun/src/kernel/cpu/ReshapeLayer.h
diff options
context:
space:
mode:
Diffstat (limited to 'runtimes/neurun/src/kernel/cpu/ReshapeLayer.h')
-rw-r--r--runtimes/neurun/src/kernel/cpu/ReshapeLayer.h4
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();