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, ExternalContext *ctx)
void	run () override
Public Member Functions inherited from onert::exec::IFunction
virtual	~IFunction ()=default
virtual void	prepare ()

Detailed Description

Definition at line 35 of file GatherLayer.h.

Constructor & Destructor Documentation

GatherLayer()

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

inline

Definition at line 38 of file GatherLayer.h.

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

Member Function Documentation

configure()

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

Definition at line 33 of file GatherLayer.cc.

{
  _input = input;
  _indices = indices;
  _axis = axis;
  _output = output;
  _ctx = ctx;
 
  if (_input->data_type() == OperandType::QUANT_GGML_Q4_0)
    ctx->initGgmlContext();
}

References onert::backend::IPortableTensor::data_type(), and onert::backend::cpu::ExternalContext::initGgmlContext().

run()

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

overridevirtual

Implements onert::exec::IFunction.

Definition at line 124 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::QUANT_GGML_Q4_0:
      runByGGMLQuantInputType();
      break;
    default:
      throw std::runtime_error("Gather: unsupported input data type");
  }
}

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

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

---

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