#include <OneHot.h>

Collaboration diagram for luci_interpreter::kernels::OneHot:

[legend]

Public Member Functions
	OneHot (const Tensor indices, const Tensor depth, const Tensor on_value, const Tensor off_value, Tensor *output, const OneHotParams &params)

const Tensor *	indices () const

const Tensor *	depth () const

const Tensor *	on_value () const

const Tensor *	off_value () const

Tensor *	output () const

void	configure () override

void	execute () const override

	OneHot (const Tensor indices, const Tensor depth, const Tensor on_value, const Tensor off_value, Tensor *output, const OneHotParams &params)

const Tensor *	indices () const

const Tensor *	depth () const

const Tensor *	on_value () const

const Tensor *	off_value () const

Tensor *	output () const

void	configure () override

void	execute () const override

Public Member Functions inherited from luci_interpreter::KernelWithParams< OneHotParams >
const OneHotParams &	params () const

Public Member Functions inherited from luci_interpreter::Kernel
virtual	~Kernel ()=default

const std::vector< const Tensor * > &	getInputTensors () const

const std::vector< Tensor * > &	getOutputTensors () const

Additional Inherited Members
Protected Member Functions inherited from luci_interpreter::KernelWithParams< OneHotParams >
	KernelWithParams (std::vector< const Tensor * > inputs, std::vector< Tensor * > outputs, const OneHotParams &params)

Protected Member Functions inherited from luci_interpreter::Kernel
	Kernel (std::vector< const Tensor * > inputs, std::vector< Tensor * > outputs)

Protected Attributes inherited from luci_interpreter::KernelWithParams< OneHotParams >
const OneHotParams	_params

Protected Attributes inherited from luci_interpreter::Kernel
const std::vector< const Tensor * >	_inputs

const std::vector< Tensor * >	_outputs

Detailed Description

Definition at line 28 of file OneHot.h.

Constructor & Destructor Documentation

◆ OneHot() [1/2]

luci_interpreter::kernels::OneHot::OneHot	(	const Tensor *	indices,
		const Tensor *	depth,
		const Tensor *	on_value,
		const Tensor *	off_value,
		Tensor *	output,
		const OneHotParams &	params
	)

Definition at line 69 of file OneHot.cpp.

  : KernelWithParams<OneHotParams>({indices, depth, on_value, off_value}, {output}, params)
{
  // Do nothing
}

References depth(), indices(), off_value(), and on_value().

◆ OneHot() [2/2]

luci_interpreter::kernels::OneHot::OneHot	(	const Tensor *	indices,
		const Tensor *	depth,
		const Tensor *	on_value,
		const Tensor *	off_value,
		Tensor *	output,
		const OneHotParams &	params
	)

Member Function Documentation

◆ configure() [1/2]

void luci_interpreter::kernels::OneHot::configure ( )

overridevirtual

Implements luci_interpreter::Kernel.

Definition at line 76 of file OneHot.cpp.

{
  // check types
  LUCI_INTERPRETER_CHECK(indices()->element_type() == DataType::S32);
  LUCI_INTERPRETER_CHECK(depth()->element_type() == DataType::S32);
  LUCI_INTERPRETER_CHECK(on_value()->element_type() == off_value()->element_type());
  LUCI_INTERPRETER_CHECK(output()->element_type() == on_value()->element_type());
 
  // check shape dependent parameters
  LUCI_INTERPRETER_CHECK(on_value()->shape().num_elements() == 1);
  LUCI_INTERPRETER_CHECK(off_value()->shape().num_elements() == 1);
  LUCI_INTERPRETER_CHECK(depth()->shape().num_elements() == 1);
  LUCI_INTERPRETER_CHECK(params().axis >= -1 && params().axis <= indices()->shape().num_dims());
 
  // define parameters that affect the output shape
  auto const depth_value = getTensorData<int32_t>(depth())[0];
  auto const &input_shape = indices()->shape();
  auto const input_dims = input_shape.num_dims();
  auto const axis = params().axis == -1 ? input_dims : params().axis;
 
  // define output shape
  Shape output_shape(input_shape.num_dims() + 1);
  {
    for (int32_t d = 0; d < axis; ++d)
      output_shape.dim(d) = input_shape.dim(d);
 
    output_shape.dim(axis) = depth_value;
 
    for (int32_t d = axis + 1; d < output_shape.num_dims(); ++d)
      output_shape.dim(d) = input_shape.dim(d - 1);
  }
 
  // reshape output
  output()->resize(output_shape);
}

References luci_interpreter::OneHotParams::axis, depth(), indices(), LUCI_INTERPRETER_CHECK, luci::must_cast(), luci_interpreter::Shape::num_dims(), off_value(), on_value(), output(), output_shape, luci_interpreter::KernelWithParams< OneHotParams >::params(), luci_interpreter::Tensor::resize(), and luci_interpreter::Tensor::shape().

◆ configure() [2/2]

void luci_interpreter::kernels::OneHot::configure ( )

overridevirtual

Implements luci_interpreter::Kernel.

◆ depth() [1/2]

const Tensor * luci_interpreter::kernels::OneHot::depth ( ) const

inline

Definition at line 35 of file OneHot.h.

35{ return _inputs[1]; }

luci_interpreter::Kernel::_inputs

const std::vector< const Tensor * > _inputs

Definition Kernel.h:52

References luci_interpreter::Kernel::_inputs.

Referenced by configure(), execute(), and OneHot().

◆ depth() [2/2]

const Tensor * luci_interpreter::kernels::OneHot::depth ( ) const

inline

Definition at line 35 of file OneHot.h.

35{ return _inputs[1]; }

References luci_interpreter::Kernel::_inputs.

◆ execute() [1/2]

void luci_interpreter::kernels::OneHot::execute ( ) const

overridevirtual

Implements luci_interpreter::Kernel.

Definition at line 112 of file OneHot.cpp.

{
  auto const depth_value = getTensorData<int32_t>(depth())[0];
  auto const axis = params().axis;
 
  switch (output()->element_type())
  {
    case loco::DataType::FLOAT32:
      OneHotComputeImpl<float>(indices(), on_value(), off_value(), depth_value, axis, output());
      break;
    case loco::DataType::U8:
      OneHotComputeImpl<uint8_t>(indices(), on_value(), off_value(), depth_value, axis, output());
      break;
    case loco::DataType::S16:
      OneHotComputeImpl<int16_t>(indices(), on_value(), off_value(), depth_value, axis, output());
      break;
    default:
      // TODO Support other data types
      throw std::runtime_error("Not supported, yet!");
      break;
  }
}