commit c6fcfb4adc37a6cf09472168dc177234d4fabdfa
author Sang-Hoon Park <sang-hoon.park@arm.com> Wed Mar 31 15:18:16 2021 +0100
committer Georgios Pinitas <georgios.pinitas@arm.com> Mon Apr 19 07:35:54 2021 +0000
tree b67afd4c8d1594053395394b24406334e66e0791
parent fe56edb4fd7a620fea4b6002d87a9763bdf8791a

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>

src/cpu/CpuTensor.cpp

diff --git a/src/cpu/CpuTensor.cpp b/src/cpu/CpuTensor.cpp
index 79dc812..6dd6d9c 100644
--- a/src/cpu/CpuTensor.cpp
+++ b/src/cpu/CpuTensor.cpp
@@ -72,7 +72,7 @@
     return bool(st) ? StatusCode::Success : StatusCode::RuntimeError;
 }
 
-arm_compute::ITensor *CpuTensor::tensor()
+arm_compute::ITensor *CpuTensor::tensor() const
 {
     return _legacy_tensor.get();
 }

src/cpu/CpuTensor.h

diff --git a/src/cpu/CpuTensor.h b/src/cpu/CpuTensor.h
index a46f1a2..b078774 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 @@
     // 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: