ArmNN
NotReleased
|
#include <CpuTensorHandle.hpp>
Public Member Functions | |
ScopedCpuTensorHandle (const TensorInfo &tensorInfo) | |
ScopedCpuTensorHandle (const ConstTensor &tensor) | |
ScopedCpuTensorHandle (const ConstCpuTensorHandle &tensorHandle) | |
ScopedCpuTensorHandle (const ScopedCpuTensorHandle &other) | |
ScopedCpuTensorHandle & | operator= (const ScopedCpuTensorHandle &other) |
~ScopedCpuTensorHandle () | |
virtual void | Allocate () override |
Public Member Functions inherited from CpuTensorHandle | |
template<typename T > | |
T * | GetTensor () const |
template<> | |
void * | GetTensor () const |
template<> | |
void * | GetTensor () const |
Public Member Functions inherited from ConstCpuTensorHandle | |
template<typename T > | |
const T * | GetConstTensor () const |
const TensorInfo & | GetTensorInfo () const |
virtual void | Manage () override |
virtual ITensorHandle * | GetParent () const override |
virtual const void * | Map (bool) const override |
virtual void | Unmap () const override |
Unmap the tensor data. More... | |
TensorShape | GetStrides () const override |
TensorShape | GetShape () const override |
template<> | |
const void * | GetConstTensor () const |
template<> | |
const void * | GetConstTensor () const |
Public Member Functions inherited from ITensorHandle | |
virtual | ~ITensorHandle () |
void * | Map (bool blocking=true) |
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) |
Additional Inherited Members | |
Protected Member Functions inherited from CpuTensorHandle | |
CpuTensorHandle (const TensorInfo &tensorInfo) | |
void | SetMemory (void *mem) |
Protected Member Functions inherited from ConstCpuTensorHandle | |
ConstCpuTensorHandle (const TensorInfo &tensorInfo) | |
void | SetConstMemory (const void *mem) |
Definition at line 106 of file CpuTensorHandle.hpp.
|
explicit |
Definition at line 54 of file CpuTensorHandle.cpp.
|
explicit |
Definition at line 59 of file CpuTensorHandle.cpp.
References BaseTensor< MemoryType >::GetMemoryArea(), and BaseTensor< MemoryType >::GetNumBytes().
|
explicit |
Definition at line 65 of file CpuTensorHandle.cpp.
References ConstCpuTensorHandle::GetConstTensor(), TensorInfo::GetNumBytes(), and ConstCpuTensorHandle::GetTensorInfo().
ScopedCpuTensorHandle | ( | const ScopedCpuTensorHandle & | other | ) |
Definition at line 71 of file CpuTensorHandle.cpp.
Definition at line 85 of file CpuTensorHandle.cpp.
|
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 90 of file CpuTensorHandle.cpp.
References TensorInfo::GetNumBytes(), CpuTensorHandle::GetTensor(), ConstCpuTensorHandle::GetTensorInfo(), and CpuTensorHandle::SetMemory().
ScopedCpuTensorHandle & operator= | ( | const ScopedCpuTensorHandle & | other | ) |