ONE - On-device Neural Engine
Loading...
Searching...
No Matches
onert::backend::train::LayerScopeTensor Class Referencefinal

#include <LayerScopeTensor.h>

Collaboration diagram for onert::backend::train::LayerScopeTensor:

Public Member Functions

 LayerScopeTensor ()=delete
 
 LayerScopeTensor (const ir::OperandInfo &info, LayerScopeTensorLifeTime lt)
 
 LayerScopeTensor (const ir::OperandInfo &info)
 
LayerScopeTensorLifeTime lifetime () const
 
- Public Member Functions inherited from onert::backend::basic::Tensor
 Tensor ()=delete
 
virtual ~Tensor ()
 
 Tensor (const ir::OperandInfo &info, DynamicMemoryManager *dynamic_mem_mgr)
 
void setBuffer (uint8_t *buffer)
 Set the Buffer object. This method is called for static and non-const tensor.
 
void setBuffer (const std::shared_ptr< Allocator > &alloc)
 Set the Buffer object. This method is called for dynamic or const tensor.
 
void deallocBuffer () override
 Reset the buffer and deallocate the allocation if it is managed by itself.
 
uint8_t * buffer () const override
 
void set_dynamic () override
 set this tensor dynamic
 
bool applyShape (const ir::Shape &new_shape) override
 Set the shape to shape and possibly re-allocate the buffer.
 
virtual void increase_ref ()
 
virtual void decrease_ref ()
 
virtual void reset_ref ()
 Reset reference count to zero and release data.
 
virtual int32_t num_references ()
 
void setShape (const ir::Shape &new_shape) override
 Set the shape of tenser to new_shape.
 
- Public Member Functions inherited from onert::backend::IPortableTensor
 IPortableTensor (const ir::OperandInfo &info)
 
virtual ~IPortableTensor ()
 
const ir::OperandInfoget_info () const
 
const ir::Sparsitysparsity () const
 
size_t total_size () const override final
 
size_t calcOffset (const ir::Coordinates &coords) const override final
 
ir::DataType data_type () const override final
 
float data_scale () const override final
 
int32_t data_zero_point () const override final
 
const std::vector< float > & data_scales () const override final
 
const std::vector< int32_t > & data_zero_points () const override
 
bool is_constant () const override final
 Return true if the tensor is constant.
 
bool is_dynamic () const override final
 Return true if the tensor needs dynamic allocation, meaning that during compile-time the outpus shape cannot be known and the output shape is calculated during kernel execution-time.
 
ir::Shape getShape () const override final
 Get ir::Shape of tensor.
 
bool has_padding () const final
 
void access (const std::function< void(ITensor &tensor)> &fn) final
 
- Public Member Functions inherited from onert::backend::ITensor
virtual ~ITensor ()
 
virtual bool is_subtensor () const
 
virtual bool needMemoryMap () const
 
virtual void enqueueWriteBuffer (const void *, bool)
 
virtual void enqueueReadBuffer (void *, bool)
 

Additional Inherited Members

- Protected Attributes inherited from onert::backend::basic::Tensor
uint8_t * _buffer
 
size_t _size
 
int32_t _num_references
 
DynamicMemoryManager_dynamic_mem_mgr
 
- Protected Attributes inherited from onert::backend::IPortableTensor
ir::OperandInfo _info
 

Detailed Description

Definition at line 37 of file LayerScopeTensor.h.

Constructor & Destructor Documentation

◆ LayerScopeTensor() [1/3]

onert::backend::train::LayerScopeTensor::LayerScopeTensor ( )
delete

◆ LayerScopeTensor() [2/3]

onert::backend::train::LayerScopeTensor::LayerScopeTensor ( const ir::OperandInfo info,
LayerScopeTensorLifeTime  lt 
)
inline

Definition at line 43 of file LayerScopeTensor.h.

44 : basic::Tensor(info, nullptr), _lifetime(lt)
45 {
46 // DO NOTHING
47 }
volatile const char info[]

◆ LayerScopeTensor() [3/3]

onert::backend::train::LayerScopeTensor::LayerScopeTensor ( const ir::OperandInfo info)
inline

Definition at line 49 of file LayerScopeTensor.h.

50 : basic::Tensor(info, nullptr), _lifetime(LayerScopeTensorLifeTime::BACKWARD)
51 {
52 // DO NOTHING
53 }

Member Function Documentation

◆ lifetime()

LayerScopeTensorLifeTime onert::backend::train::LayerScopeTensor::lifetime ( ) const
inline

Definition at line 55 of file LayerScopeTensor.h.

55{ return _lifetime; }

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