ArmNN
 25.02
All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
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 416 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

Definition at line 37 of file NeonTensorHandle.cpp.

38 {
39  return &m_TensorInfo;
40 }

◆ 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: