onert::backend::cpu::ops::OneHotLayer Class Reference

#include <OneHotLayer.h>

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

Public Member Functions
	OneHotLayer ()
template<typename T >
void	oneHotImpl ()
void	configure (const IPortableTensor *indices, const IPortableTensor *depth, const IPortableTensor *on_value, const IPortableTensor *off_value, 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 33 of file OneHotLayer.h.

Constructor & Destructor Documentation

OneHotLayer()

onert::backend::cpu::ops::OneHotLayer::OneHotLayer ( )

inline

Definition at line 36 of file OneHotLayer.h.

    : _indices(nullptr), _depth(nullptr), _on_value(nullptr), _off_value(nullptr), _output(nullptr),
      _axis(-1)
  {
    // DO NOTHING
  }

Member Function Documentation

configure()

void onert::backend::cpu::ops::OneHotLayer::configure	(	const IPortableTensor *	indices,
		const IPortableTensor *	depth,
		const IPortableTensor *	on_value,
		const IPortableTensor *	off_value,
		IPortableTensor *	output,
		int32_t	axis
	)

Definition at line 40 of file OneHotLayer.cc.

{
  _indices = indices;
  _output = output;
  _depth = depth;
  _on_value = on_value;
  _off_value = off_value;
  _axis = axis;
}

oneHotImpl()

template<typename T >

void onert::backend::cpu::ops::OneHotLayer::oneHotImpl

(

)

Definition at line 32 of file OneHotLayer.cc.

{
  // It assumes index is int32_t type.
  nnfw::cker::OneHot<T, int32_t>(
    *getBuffer<int32_t>(_depth), *getBuffer<T>(_on_value), *getBuffer<T>(_off_value), _axis,
    getShape(_indices), getBuffer<int32_t>(_indices), getShape(_output), getBuffer<T>(_output));
}

References onert::backend::cpu::ops::getShape().

run()

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

overridevirtual

Implements onert::exec::IFunction.

Definition at line 52 of file OneHotLayer.cc.

{
  if (_output->data_type() == OperandType::FLOAT32)
  {
    oneHotImpl<float>();
  }
  else
  {
    throw std::runtime_error{"OneHot: unsupported 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/OneHotLayer.h
• runtime/onert/backend/cpu/ops/OneHotLayer.cc