#include <RangeLayer.h>

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

Public Member Functions
	RangeLayer ()

void	configure (const IPortableTensor start, const IPortableTensor limit, const IPortableTensor delta, IPortableTensor output)

void	run () override

Public Member Functions inherited from onert::exec::IFunction
virtual	~IFunction ()=default

virtual void	prepare ()

Detailed Description

Definition at line 26 of file RangeLayer.h.

Constructor & Destructor Documentation

◆ RangeLayer()

onert::backend::cpu::ops::RangeLayer::RangeLayer ( )

Definition at line 25 of file RangeLayer.cc.

                       : _start(nullptr), _limit(nullptr), _delta(nullptr), _output(nullptr)
{
  // DO NOTHING
}

Member Function Documentation

◆ configure()

void onert::backend::cpu::ops::RangeLayer::configure	(	const IPortableTensor *	start,
		const IPortableTensor *	limit,
		const IPortableTensor *	delta,
		IPortableTensor *	output
	)

Definition at line 30 of file RangeLayer.cc.

{
  _start = start;
  _limit = limit;
  _delta = delta;
  _output = output;
}

◆ run()

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

overridevirtual

Implements onert::exec::IFunction.

Definition at line 39 of file RangeLayer.cc.

{
  switch (_output->data_type())
  {
    case OperandType::FLOAT32:
      nnfw::cker::Range<float>(getBuffer<float>(_start), getBuffer<float>(_limit),
                               getBuffer<float>(_delta), getBuffer<float>(_output));
      break;
    case OperandType::INT32:
      nnfw::cker::Range<int32_t>(getBuffer<int32_t>(_start), getBuffer<int32_t>(_limit),
                                 getBuffer<int32_t>(_delta), getBuffer<int32_t>(_output));
      break;
    default:
      throw std::runtime_error{"Range: unsupported data type"};
      break;
  }
}

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

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