onert::backend::cpu::ops::RangeLayer Class Reference

#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 32 of file RangeLayer.h.

Constructor & Destructor Documentation

RangeLayer()

onert::backend::cpu::ops::RangeLayer::RangeLayer

(

)

Definition at line 31 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 36 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 45 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().

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

---

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

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