diff options
Diffstat (limited to 'src/gpu')
-rw-r--r-- | src/gpu/cl/ClTensor.cpp | 2 | ||||
-rw-r--r-- | src/gpu/cl/ClTensor.h | 4 |
2 files changed, 3 insertions, 3 deletions
diff --git a/src/gpu/cl/ClTensor.cpp b/src/gpu/cl/ClTensor.cpp index db2081c4ed..0df07813e3 100644 --- a/src/gpu/cl/ClTensor.cpp +++ b/src/gpu/cl/ClTensor.cpp @@ -83,7 +83,7 @@ StatusCode ClTensor::import(void *handle, ImportMemoryType type) return StatusCode::Success; } -arm_compute::ITensor *ClTensor::tensor() +arm_compute::ITensor *ClTensor::tensor() const { return _legacy_tensor.get(); } diff --git a/src/gpu/cl/ClTensor.h b/src/gpu/cl/ClTensor.h index 4188f622d6..99d228c0b8 100644 --- a/src/gpu/cl/ClTensor.h +++ b/src/gpu/cl/ClTensor.h @@ -24,7 +24,7 @@ #ifndef SRC_GPU_CLTENSOR_H #define SRC_GPU_CLTENSOR_H -#include "src/common/ITensor.h" +#include "src/common/ITensorV2.h" #include "arm_compute/runtime/CL/CLTensor.h" @@ -53,7 +53,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: |