ArmNN
NotReleased
|
#include <CpuTensorHandle.hpp>
Public Member Functions | |
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 () |
virtual void | Allocate ()=0 |
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) |
Protected Member Functions | |
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 76 of file CpuTensorHandle.hpp.
|
protected |
Definition at line 42 of file CpuTensorHandle.cpp.
void * GetTensor | ( | ) | const |
Definition at line 49 of file CpuTensorHandle.cpp.
|
inline |
Definition at line 80 of file CpuTensorHandle.hpp.
References ConstCpuTensorHandle::GetTensorInfo().
Referenced by ScopedCpuTensorHandle::Allocate(), and ConstPassthroughCpuTensorHandle::ConstPassthroughCpuTensorHandle().
void* GetTensor | ( | ) | const |
|
inlineprotected |
Definition at line 89 of file CpuTensorHandle.hpp.
References ConstCpuTensorHandle::SetConstMemory().
Referenced by ScopedCpuTensorHandle::Allocate(), and ScopedCpuTensorHandle::operator=().