onert::backend::cpu::ops::TopKV2Layer Class Reference

#include <TopKV2Layer.h>

Collaboration diagram for onert::backend::cpu::ops::TopKV2Layer:

Public Member Functions
	TopKV2Layer ()
void	configure (const IPortableTensor *input, IPortableTensor *output_value, IPortableTensor *output_indices, const int32_t k)
void	run () override
Public Member Functions inherited from onert::exec::IFunction
virtual	~IFunction ()=default
virtual void	prepare ()

Detailed Description

Definition at line 27 of file TopKV2Layer.h.

Constructor & Destructor Documentation

TopKV2Layer()

onert::backend::cpu::ops::TopKV2Layer::TopKV2Layer

(

)

Definition at line 56 of file TopKV2Layer.cc.

  : _input(nullptr), _output_value(nullptr), _output_indices(nullptr), _k(0)
{
}

Member Function Documentation

configure()

void onert::backend::cpu::ops::TopKV2Layer::configure	(	const IPortableTensor *	input,
		IPortableTensor *	output_value,
		IPortableTensor *	output_indices,
		const int32_t	k
	)

Definition at line 61 of file TopKV2Layer.cc.

{
  _input = input;
  _output_value = output_value;
  _output_indices = output_indices;
 
  _k = static_cast<uint32_t>(k);
}

run()

void onert::backend::cpu::ops::TopKV2Layer::run ( )

overridevirtual

Implements onert::exec::IFunction.

Definition at line 71 of file TopKV2Layer.cc.

{
  if (_output_indices->data_type() != OperandType::INT32)
    throw std::runtime_error{"TopKV2: unsupported output indices type"};
 
  if (_input->data_type() == OperandType::FLOAT32)
  {
    nnfw::cker::TopKV2<float, int32_t>(getShape(_input), getBuffer<float>(_input), _k,
                                       getBuffer<float>(_output_value),
                                       getBuffer<int32_t>(_output_indices));
  }
  else if (_input->data_type() == OperandType::INT32)
  {
    nnfw::cker::TopKV2<int32_t, int32_t>(getShape(_input), getBuffer<int32_t>(_input), _k,
                                         getBuffer<int32_t>(_output_value),
                                         getBuffer<int32_t>(_output_indices));
  }
  else if (_input->data_type() == OperandType::QUANT_UINT8_ASYMM)
  {
    nnfw::cker::TopKV2<uint8_t, int32_t>(getShape(_input), getBuffer<uint8_t>(_input), _k,
                                         getBuffer<uint8_t>(_output_value),
                                         getBuffer<int32_t>(_output_indices));
  }
  else
  {
    throw std::runtime_error{"TopKV2: unsupported data type"};
  }
}

References onert::backend::IPortableTensor::data_type(), and onert::backend::cpu::ops::getShape().

---

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

• runtime/onert/backend/cpu/ops/TopKV2Layer.h
• runtime/onert/backend/cpu/ops/TopKV2Layer.cc