A tensor class that can be trained. More...

#include <ITrainableTensor.h>

Collaboration diagram for onert::backend::train::ITrainableTensor:

[legend]

Public Member Functions
virtual	~ITrainableTensor ()=default

virtual std::vector< ITensor * >	optVars ()=0
	Get optimizer variables of this trainable tensor.

	IPortableTensor (const ir::OperandInfo &info)

Public Member Functions inherited from onert::backend::IPortableTensor
	IPortableTensor (const ir::OperandInfo &info)

virtual	~IPortableTensor ()

const ir::OperandInfo &	get_info () const

const ir::Sparsity *	sparsity () 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 uint8_t *	buffer () const =0

virtual bool	applyShape (const ir::Shape &)
	Set the shape to `shape` and possibly re-allocate the buffer.

virtual void	set_dynamic ()
	set this tensor dynamic

virtual void	deallocBuffer ()
	Dealloc the buffer (only for dynamic tensors)

virtual void	setShape (const ir::Shape &)
	Set the shape of tenser to new_shape.

virtual bool	is_subtensor () const

virtual bool	needMemoryMap () const

virtual void	enqueueWriteBuffer (const void *, bool)

virtual void	enqueueReadBuffer (void *, bool)