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