ArmNN
 21.02
sdb Namespace Reference

Classes

class  SampleDynamicAdditionWorkload
 
class  SampleDynamicLayerSupport
 
class  SampleDynamicTensorHandleFactory
 
class  SampleDynamicWorkloadFactory
 
class  SampleMemoryManager
 
class  SampleTensorHandle
 

Typedefs

using FactoryId = armnn::ITensorHandleFactory::FactoryId
 

Functions

const armnn::TensorInfoGetTensorInfo (const armnn::ITensorHandle *tensorHandle)
 
const float * GetInputTensorData (unsigned int idx, const armnn::AdditionQueueDescriptor &data)
 
float * GetOutputTensorData (unsigned int idx, const armnn::AdditionQueueDescriptor &data)
 
constexpr const char * SampleDynamicBackendId ()
 
constexpr const char * SampleDynamicTensorHandleFactoryId ()
 

Typedef Documentation

◆ FactoryId

Function Documentation

◆ GetInputTensorData()

const float* sdb::GetInputTensorData ( unsigned int  idx,
const armnn::AdditionQueueDescriptor data 
)

Definition at line 22 of file SampleDynamicAdditionWorkload.cpp.

References QueueDescriptor::m_Inputs, and ITensorHandle::Map().

Referenced by SampleDynamicAdditionWorkload::Execute().

23 {
24  const armnn::ITensorHandle* tensorHandle = data.m_Inputs[idx];
25  return reinterpret_cast<const float*>(tensorHandle->Map());
26 }
virtual const void * Map(bool blocking=true) const =0
Map the tensor data for access.
std::vector< ITensorHandle * > m_Inputs

◆ GetOutputTensorData()

float* sdb::GetOutputTensorData ( unsigned int  idx,
const armnn::AdditionQueueDescriptor data 
)

Definition at line 28 of file SampleDynamicAdditionWorkload.cpp.

References QueueDescriptor::m_Outputs, and ITensorHandle::Map().

Referenced by SampleDynamicAdditionWorkload::Execute().

29 {
30  armnn::ITensorHandle* tensorHandle = data.m_Outputs[idx];
31  return reinterpret_cast<float*>(tensorHandle->Map());
32 }
virtual const void * Map(bool blocking=true) const =0
Map the tensor data for access.
std::vector< ITensorHandle * > m_Outputs

◆ GetTensorInfo()

const armnn::TensorInfo& sdb::GetTensorInfo ( const armnn::ITensorHandle tensorHandle)
inline

Definition at line 14 of file SampleDynamicAdditionWorkload.cpp.

References SampleTensorHandle::GetTensorInfo().

Referenced by SampleDynamicAdditionWorkload::Execute().

15 {
16  // We know that reference workloads use RefTensorHandles for inputs and outputs
17  const SampleTensorHandle* sampleTensorHandle =
18  static_cast<const SampleTensorHandle*>(tensorHandle);
19  return sampleTensorHandle->GetTensorInfo();
20 }

◆ SampleDynamicBackendId()

◆ SampleDynamicTensorHandleFactoryId()

constexpr const char* sdb::SampleDynamicTensorHandleFactoryId ( )

Definition at line 15 of file SampleDynamicTensorHandleFactory.hpp.

Referenced by SampleDynamicTensorHandleFactory::GetIdStatic().

15 { return "Arm/SampleDynamic/TensorHandleFactory"; }