onert::backend::builtin::train::kernel::PermuteLayer Class Reference

#include <PermuteLayer.h>

Collaboration diagram for onert::backend::builtin::train::kernel::PermuteLayer:

Public Member Functions
	PermuteLayer (const std::vector< ITensor * > &src_tensors, const std::vector< ITensor * > &dst_tensors, const std::vector< ITensor * > &input_back_prop_tensors, const std::vector< ITensor * > &output_back_prop_tensors, const std::vector< ir::PermuteType > &types, bool ignore_forward_in_training, const std::shared_ptr< ExternalContext > &external_context)
void	optimize () override
void	forward (bool training) override
void	backward () override
Public Member Functions inherited from onert::backend::builtin::kernel::PermuteLayer
	PermuteLayer (const std::vector< ITensor * > &src_tensors, const std::vector< ITensor * > &dst_tensors, const std::vector< ir::PermuteType > &types, const std::shared_ptr< ExternalContext > &external_context)
void	run () override
Public Member Functions inherited from onert::exec::IPermuteFunction
virtual void	prepare () override
Public Member Functions inherited from onert::exec::IFunction
virtual	~IFunction ()=default
Public Member Functions inherited from onert::exec::train::ITrainableFunction
virtual	~ITrainableFunction ()=default
virtual std::optional< backend::train::LayerScopeTensors >	registerLayerScopeTensors ()

Additional Inherited Members
Protected Member Functions inherited from onert::exec::IPermuteFunction
void	permute (backend::ITensor *src_tensor, backend::ITensor *dst_tensor, size_t rank, std::vector< size_t > &src_offsets, std::vector< size_t > &dst_offsets, const ir::PermuteType &permute_type)
const std::type_info &	underlying_type (ir::DataType type) const
Protected Attributes inherited from onert::exec::IPermuteFunction
std::vector< backend::ITensor * >	_src_tensors
std::vector< backend::ITensor * >	_dst_tensors
std::vector< std::vector< size_t > >	_src_tensors_offsets
std::vector< std::vector< size_t > >	_dst_tensors_offsets
std::vector< ir::PermuteType >	_permute_types
std::unordered_map< const backend::ITensor *, std::vector< uint8_t > >	_buffers_map

Detailed Description

Definition at line 27 of file PermuteLayer.h.

Constructor & Destructor Documentation

PermuteLayer()

onert::backend::builtin::train::kernel::PermuteLayer::PermuteLayer	(	const std::vector< ITensor * > &	src_tensors,
		const std::vector< ITensor * > &	dst_tensors,
		const std::vector< ITensor * > &	input_back_prop_tensors,
		const std::vector< ITensor * > &	output_back_prop_tensors,
		const std::vector< ir::PermuteType > &	types,
		bool	ignore_forward_in_training,
		const std::shared_ptr< ExternalContext > &	external_context
	)

Definition at line 24 of file PermuteLayer.cc.

  : builtin::kernel::PermuteLayer{src_tensors, dst_tensors, types, external_context},
    _input_back_prop_tensors{input_back_prop_tensors},
    _output_back_prop_tensors{output_back_prop_tensors},
    _ignore_forward_in_training{ignore_forward_in_training}
{
  assert(input_back_prop_tensors.size() == output_back_prop_tensors.size());
  assert(src_tensors.size() == dst_tensors.size());
}

Member Function Documentation

backward()

void onert::backend::builtin::train::kernel::PermuteLayer::backward ( )

overridevirtual

Implements onert::exec::train::ITrainableFunction.

Definition at line 55 of file PermuteLayer.cc.

{
  for (uint32_t i = 0; i < _output_back_prop_tensors.size(); ++i)
  {
    auto src_back_prop = _output_back_prop_tensors.at(i);
    auto dst_back_prop = _input_back_prop_tensors.at(i);
 
    // NOTE The back propagation tensors corresponding to inputs/outputs of model are nullptr
    //      because permuting those tensors is meaningless
    if (src_back_prop && dst_back_prop)
    {
      const auto rank = src_back_prop->getShape().rank();
      auto output_offsets = _dst_tensors_offsets.at(i);
      auto input_offsets = _src_tensors_offsets.at(i);
      auto permute_type = _permute_types.at(i);
 
      exec::IPermuteFunction::permute(src_back_prop, dst_back_prop, rank, output_offsets,
                                      input_offsets, permute_type);
    }
  }
}

References onert::exec::IPermuteFunction::_dst_tensors_offsets, onert::exec::IPermuteFunction::_permute_types, onert::exec::IPermuteFunction::_src_tensors_offsets, and onert::exec::IPermuteFunction::permute().

forward()

void onert::backend::builtin::train::kernel::PermuteLayer::forward ( bool  training )

overridevirtual

Implements onert::exec::train::ITrainableFunction.

Definition at line 47 of file PermuteLayer.cc.

{
  if (_ignore_forward_in_training)
    return;
 
  builtin::kernel::PermuteLayer::run();
}

References onert::backend::builtin::kernel::PermuteLayer::run().

optimize()

void onert::backend::builtin::train::kernel::PermuteLayer::optimize ( )

overridevirtual

Reimplemented from onert::backend::builtin::kernel::PermuteLayer.

Definition at line 40 of file PermuteLayer.cc.

{
  builtin::kernel::PermuteLayer::optimize();
 
  // TODO Calculate offsets of back propagation tensors if necessary
}

References onert::backend::builtin::kernel::PermuteLayer::optimize().

---

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

• runtime/onert/core/src/backend/builtin/train/kernel/PermuteLayer.h
• runtime/onert/core/src/backend/builtin/train/kernel/PermuteLayer.cc