diff options
| author | Sang-Hoon Park <sang-hoon.park@arm.com> | 2021-03-31 15:18:16 +0100 |
|---|---|---|
| committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2021-04-19 07:35:54 +0000 |
| commit | c6fcfb4adc37a6cf09472168dc177234d4fabdfa (patch) | |
| tree | b67afd4c8d1594053395394b24406334e66e0791 /src/cpu | |
| parent | fe56edb4fd7a620fea4b6002d87a9763bdf8791a (diff) | |
| download | ComputeLibrary-c6fcfb4adc37a6cf09472168dc177234d4fabdfa.tar.gz | |
Add Tensor related utilities to the new API
A couple of utility functions to get the information
about tensors are added. Those functions are placed
at an additional header file for better grouping.
Related test cases are also added.
Resolves: COMPMID-4376
Change-Id: I6bd09cbf60fddcf4fe651906982397afb0451392
Signed-off-by: Sang-Hoon Park <sang-hoon.park@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5405
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
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: |