diff options
Diffstat (limited to 'src/cpu')
-rw-r--r-- | src/cpu/CpuTensor.cpp | 2 | ||||
-rw-r--r-- | src/cpu/CpuTensor.h | 4 |
2 files changed, 3 insertions, 3 deletions
diff --git a/src/cpu/CpuTensor.cpp b/src/cpu/CpuTensor.cpp index 79dc812c58..6dd6d9c31b 100644 --- a/src/cpu/CpuTensor.cpp +++ b/src/cpu/CpuTensor.cpp @@ -72,7 +72,7 @@ StatusCode CpuTensor::import(void *handle, ImportMemoryType type) return bool(st) ? StatusCode::Success : StatusCode::RuntimeError; } -arm_compute::ITensor *CpuTensor::tensor() +arm_compute::ITensor *CpuTensor::tensor() const { return _legacy_tensor.get(); } 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: |