ArmNN
 24.02
NeonTensorDecorator Class Reference

NeonTensorDecorator wraps an existing Neon tensor allowing us to override the TensorInfo for it. More...

#include <NeonTensorHandle.hpp>

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

Public Member Functions

 NeonTensorDecorator ()
 
 NeonTensorDecorator (arm_compute::ITensor *original, const TensorInfo &info)
 
 ~NeonTensorDecorator ()=default
 
 NeonTensorDecorator (const NeonTensorDecorator &)=delete
 
NeonTensorDecoratoroperator= (const NeonTensorDecorator &)=delete
 
 NeonTensorDecorator (NeonTensorDecorator &&)=default
 
NeonTensorDecoratoroperator= (NeonTensorDecorator &&)=default
 
arm_compute::ITensorInfo * info () const override
 
arm_compute::ITensorInfo * info () override
 
uint8_t * buffer () const override
 

Detailed Description

NeonTensorDecorator wraps an existing Neon tensor allowing us to override the TensorInfo for it.

Definition at line 409 of file NeonTensorHandle.hpp.

Constructor & Destructor Documentation

◆ NeonTensorDecorator() [1/4]

Definition at line 20 of file NeonTensorHandle.cpp.

21  : m_Original(nullptr), m_TensorInfo()
22 {
23 }

◆ NeonTensorDecorator() [2/4]

NeonTensorDecorator ( arm_compute::ITensor *  original,
const TensorInfo info 
)

Definition at line 25 of file NeonTensorHandle.cpp.

26  : m_Original(nullptr), m_TensorInfo()
27 {
28  m_TensorInfo = armcomputetensorutils::BuildArmComputeTensorInfo(tensorInfo);
29  m_Original = parent;
30 }

◆ ~NeonTensorDecorator()

~NeonTensorDecorator ( )
default

◆ NeonTensorDecorator() [3/4]

◆ NeonTensorDecorator() [4/4]

Member Function Documentation

◆ buffer()

uint8_t * buffer ( ) const
override

Definition at line 42 of file NeonTensorHandle.cpp.

43 {
44  return m_Original->buffer();
45 }

Referenced by NeonTensorHandleDecorator::Map().

◆ info() [1/2]

arm_compute::ITensorInfo * info ( ) const
override

Definition at line 32 of file NeonTensorHandle.cpp.

33 {
34  return &m_TensorInfo;
35 }

Referenced by NeonTensorHandleDecorator::GetDataType(), and NeonTensorHandleDecorator::Map().

◆ info() [2/2]

arm_compute::ITensorInfo* info ( )
override

◆ operator=() [1/2]

NeonTensorDecorator& operator= ( const NeonTensorDecorator )
delete

◆ operator=() [2/2]

NeonTensorDecorator& operator= ( NeonTensorDecorator &&  )
default

The documentation for this class was generated from the following files: