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/gpu | |
| 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/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: |