ArmNN
NotReleased
|
#include <NeonTensorHandle.hpp>
Public Member Functions | |
NeonTensorHandle (const TensorInfo &tensorInfo) | |
NeonTensorHandle (const TensorInfo &tensorInfo, DataLayout dataLayout, MemorySourceFlags importFlags=static_cast< MemorySourceFlags >(MemorySource::Malloc)) | |
arm_compute::ITensor & | GetTensor () override |
arm_compute::ITensor const & | GetTensor () const override |
virtual void | Allocate () override |
virtual void | Manage () override |
virtual ITensorHandle * | GetParent () const override |
virtual arm_compute::DataType | GetDataType () const override |
virtual void | SetMemoryGroup (const std::shared_ptr< arm_compute::IMemoryGroup > &memoryGroup) 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 |
void | SetImportFlags (MemorySourceFlags importFlags) |
MemorySourceFlags | GetImportFlags () const override |
Get flags describing supported import sources. More... | |
void | SetImportEnabledFlag (bool importEnabledFlag) |
virtual bool | Import (void *memory, MemorySource source) override |
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... | |
Definition at line 24 of file NeonTensorHandle.hpp.
|
inline |
Definition at line 27 of file NeonTensorHandle.hpp.
|
inline |
Definition at line 35 of file NeonTensorHandle.hpp.
|
inlineoverridevirtual |
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 49 of file NeonTensorHandle.hpp.
|
inlineoverridevirtual |
Implements IAclTensorHandle.
Definition at line 70 of file NeonTensorHandle.hpp.
Referenced by NeonTensorHandle::Import().
|
inlineoverridevirtual |
Get flags describing supported import sources.
Reimplemented from ITensorHandle.
Definition at line 102 of file NeonTensorHandle.hpp.
|
inlineoverridevirtual |
Get the parent tensor if this is a subtensor.
Implements ITensorHandle.
Definition at line 68 of file NeonTensorHandle.hpp.
|
inlineoverridevirtual |
Get the number of elements for each dimension ordered from slowest iterating dimension to fastest iterating dimension.
Implements ITensorHandle.
Definition at line 92 of file NeonTensorHandle.hpp.
|
inlineoverridevirtual |
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.
Implements ITensorHandle.
Definition at line 87 of file NeonTensorHandle.hpp.
|
inlineoverridevirtual |
Implements IAclTensorHandle.
Definition at line 46 of file NeonTensorHandle.hpp.
|
inlineoverridevirtual |
Implements IAclTensorHandle.
Definition at line 47 of file NeonTensorHandle.hpp.
|
inlineoverridevirtual |
Import externally allocated memory
memory | base address of the memory being imported. |
source | source of the allocation for the memory being imported. |
Reimplemented from ITensorHandle.
Definition at line 112 of file NeonTensorHandle.hpp.
References NeonTensorHandle::GetDataType(), and armnn::Malloc.
|
inlineoverridevirtual |
Indicate to the memory manager that this resource is active. This is used to compute overlapping lifetimes of resources.
Implements ITensorHandle.
Definition at line 58 of file NeonTensorHandle.hpp.
|
inlineoverridevirtual |
Map the tensor data for access.
blocking | hint to block the calling thread until all other accesses are complete. (backend dependent) |
Implements ITensorHandle.
Definition at line 80 of file NeonTensorHandle.hpp.
|
inline |
Definition at line 107 of file NeonTensorHandle.hpp.
|
inline |
Definition at line 97 of file NeonTensorHandle.hpp.
|
inlineoverridevirtual |
Implements IAclTensorHandle.
Definition at line 75 of file NeonTensorHandle.hpp.
|
inlineoverridevirtual |
Unmap the tensor data.
Implements ITensorHandle.
Definition at line 85 of file NeonTensorHandle.hpp.