From 1f58f03d82c482626b1b4673b6c0e25da4338fb5 Mon Sep 17 00:00:00 2001
From: James Conroy <james.conroy@arm.com>
Date: Tue, 27 Apr 2021 17:13:27 +0100
Subject: IVGCVSW-5815 Generalise ConstCpuTensorHandle

* Generalises ConstCpuTensorHandle and inherited
  classes by removing 'Cpu' from aliases.
* New renamed classes: ConstTensorHandle, TensorHandle,
  ScopedTensorHandle, PassthroughTensorHandle,
  ConstPassthroughTensorHandle.

Signed-off-by: James Conroy <james.conroy@arm.com>
Change-Id: I1824e0e134202735fb77051f20a7252f161dfe16
---
 src/armnn/optimizations/ConvertConstants.hpp | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

(limited to 'src/armnn/optimizations/ConvertConstants.hpp')

diff --git a/src/armnn/optimizations/ConvertConstants.hpp b/src/armnn/optimizations/ConvertConstants.hpp
index df5a5b4f67..66b3d2685a 100644
--- a/src/armnn/optimizations/ConvertConstants.hpp
+++ b/src/armnn/optimizations/ConvertConstants.hpp
@@ -9,7 +9,7 @@
 
 #include <armnnUtils/FloatingPointConverter.hpp>
 
-#include <backendsCommon/CpuTensorHandle.hpp>
+#include <backendsCommon/TensorHandle.hpp>
 
 #include <armnn/utility/IgnoreUnused.hpp>
 
@@ -23,7 +23,7 @@ namespace optimizations
 
 struct BFloat16ToFloat32
 {
-    static void Func(std::shared_ptr<ConstCpuTensorHandle>& handle)
+    static void Func(std::shared_ptr<ConstTensorHandle>& handle)
     {
         const TensorInfo& info = handle->GetTensorInfo();
 
@@ -37,14 +37,14 @@ struct BFloat16ToFloat32
 
             TensorInfo newInfo(info.GetShape(), DataType::Float32);
             ConstTensor newInput(newInfo, newValues);
-            handle.reset(new ScopedCpuTensorHandle(newInput));
+            handle.reset(new ScopedTensorHandle(newInput));
         }
     }
 };
 
 struct Float16ToFloat32
 {
-    static void Func(std::shared_ptr<ConstCpuTensorHandle>& handle)
+    static void Func(std::shared_ptr<ConstTensorHandle>& handle)
     {
         const TensorInfo& info = handle->GetTensorInfo();
 
@@ -58,14 +58,14 @@ struct Float16ToFloat32
 
             TensorInfo newInfo(info.GetShape(), DataType::Float32);
             ConstTensor newInput(newInfo, newValues);
-            handle.reset(new ScopedCpuTensorHandle(newInput));
+            handle.reset(new ScopedTensorHandle(newInput));
         }
     }
 };
 
 struct Float32ToBFloat16
 {
-    static void Func(std::shared_ptr<ConstCpuTensorHandle>& handle)
+    static void Func(std::shared_ptr<ConstTensorHandle>& handle)
     {
         const TensorInfo& info = handle->GetTensorInfo();
 
@@ -79,14 +79,14 @@ struct Float32ToBFloat16
 
             TensorInfo newInfo(info.GetShape(), DataType::BFloat16);
             ConstTensor newInput(newInfo, newValues);
-            handle.reset(new ScopedCpuTensorHandle(newInput));
+            handle.reset(new ScopedTensorHandle(newInput));
         }
     }
 };
 
 struct Float32ToFloat16
 {
-    static void Func(std::shared_ptr<ConstCpuTensorHandle>& handle)
+    static void Func(std::shared_ptr<ConstTensorHandle>& handle)
     {
         const TensorInfo& info = handle->GetTensorInfo();
 
@@ -100,7 +100,7 @@ struct Float32ToFloat16
 
             TensorInfo newInfo(info.GetShape(), DataType::Float16);
             ConstTensor newInput(newInfo, newValues);
-            handle.reset(new ScopedCpuTensorHandle(newInput));
+            handle.reset(new ScopedTensorHandle(newInput));
         }
     }
 };
-- 
cgit v1.2.1