aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/ITensorHandle.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/backendsCommon/ITensorHandle.hpp')
-rw-r--r--src/backends/backendsCommon/ITensorHandle.hpp17
1 files changed, 5 insertions, 12 deletions
diff --git a/src/backends/backendsCommon/ITensorHandle.hpp b/src/backends/backendsCommon/ITensorHandle.hpp
index 02f4ed6e5a..176b021d76 100644
--- a/src/backends/backendsCommon/ITensorHandle.hpp
+++ b/src/backends/backendsCommon/ITensorHandle.hpp
@@ -12,13 +12,6 @@ class TensorShape;
class ITensorHandle
{
public:
- enum Type
- {
- Cpu,
- CL,
- Neon
- };
-
virtual ~ITensorHandle(){}
/// Indicate to the memory manager that this resource is active.
@@ -29,10 +22,6 @@ public:
/// This is used to compute overlapping lifetimes of resources.
virtual void Allocate() = 0;
- /// Get the type backend associated with the tensor handle.
- /// \return Type enum
- virtual ITensorHandle::Type GetType() const = 0;
-
/// Get the parent tensor if this is a subtensor.
/// \return a pointer to the parent tensor. Otherwise nullptr if not a subtensor.
virtual ITensorHandle* GetParent() const = 0;
@@ -64,10 +53,14 @@ public:
/// \return a TensorShape filled with the strides for each dimension
virtual TensorShape GetStrides() const = 0;
- /// Get the number of elements for each dimension orderd from slowest iterating dimension
+ /// Get the number of elements for each dimension ordered from slowest iterating dimension
/// to fastest iterating dimension.
/// \return a TensorShape filled with the number of elements for each dimension.
virtual TensorShape GetShape() const = 0;
+
+ // Testing support to be able to verify and set tensor data content
+ virtual void CopyOutTo(void* memory) const = 0;
+ virtual void CopyInFrom(const void* memory) = 0;
};
}