diff options
Diffstat (limited to 'src/cpu/CpuTensor.h')
-rw-r--r-- | src/cpu/CpuTensor.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/cpu/CpuTensor.h b/src/cpu/CpuTensor.h index a46f1a26cb..b078774c99 100644 --- a/src/cpu/CpuTensor.h +++ b/src/cpu/CpuTensor.h @@ -24,7 +24,7 @@ #ifndef SRC_CPU_CPUTENSOR_H #define SRC_CPU_CPUTENSOR_H -#include "src/common/ITensor.h" +#include "src/common/ITensorV2.h" #include "arm_compute/runtime/Tensor.h" @@ -51,7 +51,7 @@ public: // Inherrited functions overriden void *map() override; StatusCode unmap() override; - arm_compute::ITensor *tensor() override; + arm_compute::ITensor *tensor() const override; StatusCode import(void *handle, ImportMemoryType type) override; private: |