onert::backend::cpu::ops::GatherLayer Class Reference

#include <GatherLayer.h>

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

Public Member Functions
	GatherLayer ()
void	configure (const IPortableTensor *input, const IPortableTensor *indices, IPortableTensor *output, int32_t axis)
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 GatherLayer.h.

Constructor & Destructor Documentation

GatherLayer()

onert::backend::cpu::ops::GatherLayer::GatherLayer ( )

inline

Definition at line 30 of file GatherLayer.h.

                : _input{nullptr}, _indices{nullptr}, _output{nullptr}, _axis{-1}
  {
    // DO NOTHING
  }

Member Function Documentation

configure()

void onert::backend::cpu::ops::GatherLayer::configure	(	const IPortableTensor *	input,
		const IPortableTensor *	indices,
		IPortableTensor *	output,
		int32_t	axis
	)

Definition at line 68 of file GatherLayer.cc.

{
  _input = input;
  _indices = indices;
  _axis = axis;
  _output = output;
}

run()

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

overridevirtual

Implements onert::exec::IFunction.

Definition at line 108 of file GatherLayer.cc.

{
  switch (_input->data_type())
  {
    case OperandType::FLOAT32:
      runByInputType<float>();
      break;
    case OperandType::QUANT_UINT8_ASYMM:
      runByInputType<uint8_t>();
      break;
    case OperandType::INT32:
      runByInputType<int32_t>();
      break;
    case OperandType::BOOL8:
      runByInputType<bool>();
      break;
    default:
      throw std::runtime_error("Gather: unsupported input data type");
  }
}

References onert::backend::IPortableTensor::data_type().

---

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

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