From c6fcfb4adc37a6cf09472168dc177234d4fabdfa Mon Sep 17 00:00:00 2001 From: Sang-Hoon Park Date: Wed, 31 Mar 2021 15:18:16 +0100 Subject: 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 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5405 Comments-Addressed: Arm Jenkins Reviewed-by: Georgios Pinitas Reviewed-by: Michele Di Giorgio Tested-by: Arm Jenkins --- src/gpu/cl/ClTensor.cpp | 2 +- src/gpu/cl/ClTensor.h | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) (limited to 'src/gpu') 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: -- cgit v1.2.1