diff options
Diffstat (limited to 'src/backends/backendsCommon/ITensorHandle.hpp')
-rw-r--r-- | src/backends/backendsCommon/ITensorHandle.hpp | 17 |
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; }; } |