onert::exec::PermuteLayer Class Reference

#include <IPermuteFunction.h>

Collaboration diagram for onert::exec::PermuteLayer:

Public Member Functions
	PermuteLayer (const std::vector< onert::backend::ITensor * > &inputs, const std::vector< onert::backend::ITensor * > &outputs, const std::vector< ir::PermuteType > &types)
virtual	~PermuteLayer ()
void	optimize () override
Public Member Functions inherited from onert::exec::IPermuteFunction
virtual void	run () override
virtual void	prepare () override
Public Member Functions inherited from onert::exec::IFunction
virtual	~IFunction ()=default

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 251 of file IPermuteFunction.h.

Constructor & Destructor Documentation

PermuteLayer()

onert::exec::PermuteLayer::PermuteLayer ( const std::vector< onert::backend::ITensor * > &  inputs, const std::vector< onert::backend::ITensor * > &  outputs, const std::vector< ir::PermuteType > &  types  )

inline

Definition at line 254 of file IPermuteFunction.h.

  {
    assert(inputs.size() == outputs.size());
    assert(inputs.size() == types.size());
    _src_tensors = inputs;
    _dst_tensors = outputs;
    _permute_types = types;
  }

References onert::exec::IPermuteFunction::_dst_tensors, onert::exec::IPermuteFunction::_permute_types, and onert::exec::IPermuteFunction::_src_tensors.

~PermuteLayer()

virtual onert::exec::PermuteLayer::~PermuteLayer ( )

inlinevirtual

Definition at line 264 of file IPermuteFunction.h.

{}

Member Function Documentation

optimize()

void onert::exec::PermuteLayer::optimize ( )

inlineoverridevirtual

Implements onert::exec::IPermuteFunction.

Definition at line 265 of file IPermuteFunction.h.

{}

---

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

• runtime/onert/core/src/exec/IPermuteFunction.h