ArmNN
 25.11
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}

Referenced by NeonTensorDecorator(), NeonTensorDecorator(), operator=(), and operator=().

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

◆ info() [1/2]

arm_compute::ITensorInfo * info ( ) const
override

Definition at line 32 of file NeonTensorHandle.cpp.

33{
34 return &m_TensorInfo;
35}

◆ 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

References NeonTensorDecorator().

◆ operator=() [2/2]

NeonTensorDecorator & operator= ( NeonTensorDecorator && )
default

References NeonTensorDecorator().


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