onert::exec::IPermuteFunction Class Reference

#include <IPermuteFunction.h>

Collaboration diagram for onert::exec::IPermuteFunction:

Public Member Functions
virtual void	run () override
virtual void	prepare () override
virtual void	optimize ()=0
Public Member Functions inherited from onert::exec::IFunction
virtual	~IFunction ()=default

Protected Member Functions
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
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 67 of file IPermuteFunction.h.

Member Function Documentation

optimize()

virtual void onert::exec::IPermuteFunction::optimize ( )

pure virtual

Implemented in onert::backend::builtin::kernel::PermuteLayer, onert::backend::builtin::train::kernel::PermuteLayer, and onert::exec::PermuteLayer.

Referenced by prepare().

permute()

void onert::exec::IPermuteFunction::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  )

protected

Definition at line 240 of file IPermuteFunction.cc.

{
  if (src_tensor->total_size() == 0)
  {
    assert(dst_tensor->total_size() == 0);
    return;
  }
 
  assert(src_tensor != dst_tensor);
  if (underlying_type(src_tensor->data_type()) != underlying_type(dst_tensor->data_type()))
  {
    typeAwareQuantize(src_tensor, dst_tensor, permute_type);
    return;
  }
 
  switch (src_tensor->data_type())
  {
    case ir::DataType::FLOAT32:
      permute<float>(src_tensor, dst_tensor, rank, src_offsets, dst_offsets, permute_type);
      break;
    case ir::DataType::INT32:
      permute<int32_t>(src_tensor, dst_tensor, rank, src_offsets, dst_offsets, permute_type);
      break;
    case ir::DataType::UINT32:
      permute<uint32_t>(src_tensor, dst_tensor, rank, src_offsets, dst_offsets, permute_type);
      break;
    case ir::DataType::BOOL8:
    case ir::DataType::QUANT_UINT8_ASYMM:
    case ir::DataType::UINT8:
      permute<uint8_t>(src_tensor, dst_tensor, rank, src_offsets, dst_offsets, permute_type);
      break;
    case ir::DataType::QUANT_INT8_ASYMM:
    case ir::DataType::QUANT_INT8_SYMM:
      permute<int8_t>(src_tensor, dst_tensor, rank, src_offsets, dst_offsets, permute_type);
      break;
    case ir::DataType::INT64:
      permute<int64_t>(src_tensor, dst_tensor, rank, src_offsets, dst_offsets, permute_type);
      break;
    case ir::DataType::QUANT_INT16_SYMM:
      permute<int16_t>(src_tensor, dst_tensor, rank, src_offsets, dst_offsets, permute_type);
      break;
    default:
      throw std::runtime_error("IPermuteFunction: Not supported data type");
      break;
  }
}

References onert::backend::ITensor::data_type(), onert::backend::ITensor::total_size(), and underlying_type().

Referenced by onert::backend::builtin::train::kernel::PermuteLayer::backward(), and onert::backend::builtin::kernel::PermuteLayer::run().

prepare()

virtual void onert::exec::IPermuteFunction::prepare ( )

inlineoverridevirtual

Reimplemented from onert::exec::IFunction.

Definition at line 72 of file IPermuteFunction.h.

{ optimize(); }

References optimize().

run()

virtual void onert::exec::IPermuteFunction::run ( )

overridevirtual

Implements onert::exec::IFunction.

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

Referenced by package.infer.session::inference().

underlying_type()

const std::type_info & onert::exec::IPermuteFunction::underlying_type ( ir::DataType  type ) const

protected

Definition at line 290 of file IPermuteFunction.cc.

{
  switch (type)
  {
    case ir::DataType::FLOAT32:
      return typeid(float);
    case ir::DataType::INT32:
      return typeid(int32_t);
    case ir::DataType::UINT32:
      return typeid(uint32_t);
    case ir::DataType::INT64:
      return typeid(int64_t);
    case ir::DataType::BOOL8:
    case ir::DataType::QUANT_UINT8_ASYMM:
    case ir::DataType::UINT8:
      return typeid(uint8_t);
    case ir::DataType::QUANT_INT8_ASYMM:
    case ir::DataType::QUANT_INT8_SYMM:
      return typeid(int8_t);
    case ir::DataType::QUANT_INT16_SYMM:
      return typeid(int16_t);
    default:
      throw std::runtime_error("IPermuteFunction: Not supported data type");
  }
}

Referenced by onert::backend::builtin::kernel::PermuteLayer::optimize(), permute(), and onert::backend::builtin::kernel::PermuteLayer::run().

Field Documentation

_buffers_map

std::unordered_map<const backend::ITensor *, std::vector<uint8_t> > onert::exec::IPermuteFunction::_buffers_map

protected

Definition at line 247 of file IPermuteFunction.h.

Referenced by onert::backend::builtin::kernel::PermuteLayer::run().

_dst_tensors

std::vector<backend::ITensor *> onert::exec::IPermuteFunction::_dst_tensors

protected

Definition at line 243 of file IPermuteFunction.h.

Referenced by onert::backend::builtin::kernel::PermuteLayer::optimize(), onert::backend::builtin::kernel::PermuteLayer::PermuteLayer(), onert::exec::PermuteLayer::PermuteLayer(), and onert::backend::builtin::kernel::PermuteLayer::run().

_dst_tensors_offsets

std::vector<std::vector<size_t> > onert::exec::IPermuteFunction::_dst_tensors_offsets

protected

Definition at line 245 of file IPermuteFunction.h.

Referenced by onert::backend::builtin::train::kernel::PermuteLayer::backward(), onert::backend::builtin::kernel::PermuteLayer::optimize(), onert::backend::builtin::kernel::PermuteLayer::PermuteLayer(), and onert::backend::builtin::kernel::PermuteLayer::run().

_permute_types

std::vector<ir::PermuteType> onert::exec::IPermuteFunction::_permute_types

protected

Definition at line 246 of file IPermuteFunction.h.

Referenced by onert::backend::builtin::train::kernel::PermuteLayer::backward(), onert::backend::builtin::kernel::PermuteLayer::optimize(), onert::backend::builtin::kernel::PermuteLayer::PermuteLayer(), onert::exec::PermuteLayer::PermuteLayer(), and onert::backend::builtin::kernel::PermuteLayer::run().

_src_tensors

std::vector<backend::ITensor *> onert::exec::IPermuteFunction::_src_tensors

protected

Definition at line 242 of file IPermuteFunction.h.

Referenced by onert::backend::builtin::kernel::PermuteLayer::optimize(), onert::backend::builtin::kernel::PermuteLayer::PermuteLayer(), onert::exec::PermuteLayer::PermuteLayer(), and onert::backend::builtin::kernel::PermuteLayer::run().

_src_tensors_offsets

std::vector<std::vector<size_t> > onert::exec::IPermuteFunction::_src_tensors_offsets

protected

Definition at line 244 of file IPermuteFunction.h.

Referenced by onert::backend::builtin::train::kernel::PermuteLayer::backward(), onert::backend::builtin::kernel::PermuteLayer::optimize(), onert::backend::builtin::kernel::PermuteLayer::PermuteLayer(), and onert::backend::builtin::kernel::PermuteLayer::run().

---

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

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