onert::backend::cpu::ops::SliceLayer Class Reference

#include <SliceLayer.h>

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

Public Member Functions
	SliceLayer ()
void	configure (const IPortableTensor *input, const IPortableTensor *begin, const IPortableTensor *size, IPortableTensor *output)
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 SliceLayer.h.

Constructor & Destructor Documentation

SliceLayer()

onert::backend::cpu::ops::SliceLayer::SliceLayer

(

)

Definition at line 32 of file SliceLayer.cc.

                       : _input(nullptr), _begin(nullptr), _size(nullptr), _output(nullptr)
{
  // DO NOTHING
}

Member Function Documentation

configure()

void onert::backend::cpu::ops::SliceLayer::configure	(	const IPortableTensor *	input,
		const IPortableTensor *	begin,
		const IPortableTensor *	size,
		IPortableTensor *	output
	)

Definition at line 91 of file SliceLayer.cc.

{
  _input = input;
  _output = output;
  _begin = begin;
  _size = size;
}

References begin, and size.

run()

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

overridevirtual

Implements onert::exec::IFunction.

Definition at line 100 of file SliceLayer.cc.

{
  if (_input->data_type() == OperandType::FLOAT32)
  {
    sliceImpl<float>();
  }
  else if (_input->data_type() == OperandType::QUANT_UINT8_ASYMM)
  {
    sliceImpl<uint8_t>();
  }
  else
  {
    throw std::runtime_error{"Slice: 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/SliceLayer.h
• runtime/onert/backend/cpu/ops/SliceLayer.cc