ArmNN
 26.01
Loading...
Searching...
No Matches
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

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