ArmNN
 24.08
ScopedTensorHandle Class Reference

#include <TensorHandle.hpp>

Inheritance diagram for ScopedTensorHandle:
[legend]
Collaboration diagram for ScopedTensorHandle:
[legend]

Public Member Functions

 ScopedTensorHandle (const TensorInfo &tensorInfo)
 
 ScopedTensorHandle (const ConstTensor &tensor)
 
 ScopedTensorHandle (const ConstTensorHandle &tensorHandle)
 
 ScopedTensorHandle (const ScopedTensorHandle &other)
 
ScopedTensorHandleoperator= (const ScopedTensorHandle &other)
 
 ~ScopedTensorHandle ()
 
virtual void Allocate () override
 Indicate to the memory manager that this resource is no longer active. More...
 
- Public Member Functions inherited from TensorHandle
template<typename T >
T * GetTensor () const
 
template<>
void * GetTensor () const
 
template<>
void * GetTensor () const
 
- Public Member Functions inherited from ConstTensorHandle
template<typename T >
const T * GetConstTensor () const
 
const TensorInfoGetTensorInfo () const
 
virtual void Manage () override
 Indicate to the memory manager that this resource is active. More...
 
virtual ITensorHandleGetParent () const override
 Get the parent tensor if this is a subtensor. More...
 
virtual const void * Map (bool) const override
 Map the tensor data for access. More...
 
virtual void Unmap () const override
 Unmap the tensor data. More...
 
TensorShape GetStrides () const override
 Get the strides for each dimension ordered from largest to smallest where the smallest value is the same as the size of a single element in the tensor. More...
 
TensorShape GetShape () const override
 Get the number of elements for each dimension ordered from slowest iterating dimension to fastest iterating dimension. More...
 
template<>
const void * GetConstTensor () const
 
template<>
const void * GetConstTensor () const
 
- Public Member Functions inherited from ITensorHandle
virtual ~ITensorHandle ()
 
void * Map (bool blocking=true)
 Map the tensor data for access. More...
 
void Unmap ()
 Unmap the tensor data that was previously mapped with call to Map(). More...
 
virtual unsigned int GetImportFlags () const
 Get flags describing supported import sources. More...
 
virtual bool Import (void *memory, MemorySource source)
 Import externally allocated memory. More...
 
virtual bool CanBeImported (void *memory, MemorySource source)
 Implementations must determine if this memory block can be imported. More...
 
virtual void Unimport ()
 Unimport externally allocated memory. More...
 
virtual std::shared_ptr< ITensorHandleDecorateTensorHandle (const TensorInfo &tensorInfo)
 Returns a decorated version of this TensorHandle allowing us to override the TensorInfo for it. More...
 

Additional Inherited Members

- Protected Member Functions inherited from TensorHandle
 TensorHandle (const TensorInfo &tensorInfo)
 
void SetMemory (void *mem)
 
- Protected Member Functions inherited from ConstTensorHandle
 ConstTensorHandle (const TensorInfo &tensorInfo)
 
void SetConstMemory (const void *mem)
 

Detailed Description

Definition at line 115 of file TensorHandle.hpp.

Constructor & Destructor Documentation

◆ ScopedTensorHandle() [1/4]

ScopedTensorHandle ( const TensorInfo tensorInfo)
explicit

Definition at line 55 of file TensorHandle.cpp.

56 : TensorHandle(tensorInfo)
57 {
58 }

◆ ScopedTensorHandle() [2/4]

ScopedTensorHandle ( const ConstTensor tensor)
explicit

Definition at line 60 of file TensorHandle.cpp.

61 : ScopedTensorHandle(tensor.GetInfo())
62 {
63  CopyFrom(tensor.GetMemoryArea(), tensor.GetNumBytes());
64 }

References BaseTensor< MemoryType >::GetMemoryArea(), and BaseTensor< MemoryType >::GetNumBytes().

◆ ScopedTensorHandle() [3/4]

ScopedTensorHandle ( const ConstTensorHandle tensorHandle)
explicit

Definition at line 66 of file TensorHandle.cpp.

67 : ScopedTensorHandle(tensorHandle.GetTensorInfo())
68 {
69  CopyFrom(tensorHandle.GetConstTensor<void>(), tensorHandle.GetTensorInfo().GetNumBytes());
70 }

References ConstTensorHandle::GetConstTensor(), TensorInfo::GetNumBytes(), and ConstTensorHandle::GetTensorInfo().

◆ ScopedTensorHandle() [4/4]

Definition at line 72 of file TensorHandle.cpp.

73 : TensorHandle(other.GetTensorInfo())
74 {
75  CopyFrom(other);
76 }

◆ ~ScopedTensorHandle()

Definition at line 86 of file TensorHandle.cpp.

87 {
88  ::operator delete(GetTensor<void>());
89 }

Member Function Documentation

◆ Allocate()

void Allocate ( )
overridevirtual

Indicate to the memory manager that this resource is no longer active.

This is used to compute overlapping lifetimes of resources.

Implements ITensorHandle.

Definition at line 91 of file TensorHandle.cpp.

92 {
93  if (GetTensor<void>() == nullptr)
94  {
95  SetMemory(::operator new(GetTensorInfo().GetNumBytes()));
96  }
97  else
98  {
99  throw InvalidArgumentException("TensorHandle::Allocate Trying to allocate a TensorHandle"
100  "that already has allocated memory.");
101  }
102 }

References ConstTensorHandle::GetTensorInfo(), and TensorHandle::SetMemory().

◆ operator=()

ScopedTensorHandle & operator= ( const ScopedTensorHandle other)

Definition at line 78 of file TensorHandle.cpp.

79 {
80  ::operator delete(GetTensor<void>());
81  SetMemory(nullptr);
82  CopyFrom(other);
83  return *this;
84 }

References TensorHandle::SetMemory().


The documentation for this class was generated from the following files:
armnn::ScopedTensorHandle::ScopedTensorHandle
ScopedTensorHandle(const TensorInfo &tensorInfo)
Definition: TensorHandle.cpp:55
armnn::ConstTensorHandle::GetTensorInfo
const TensorInfo & GetTensorInfo() const
Definition: TensorHandle.hpp:40
armnn::TensorHandle::SetMemory
void SetMemory(void *mem)
Definition: TensorHandle.hpp:98
armnn::TensorHandle::TensorHandle
TensorHandle(const TensorInfo &tensorInfo)
Definition: TensorHandle.cpp:43