onert::backend::cpu::ops::StridedSliceLayer Class Reference

#include <StridedSliceLayer.h>

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

Public Member Functions
	StridedSliceLayer ()
void	configure (const IPortableTensor *input, const IPortableTensor *begin, const IPortableTensor *end, const IPortableTensor *strides, IPortableTensor *output, const int32_t begin_mask, const int32_t end_mask, const int32_t shrink_axis_mask)
void	run () override
Public Member Functions inherited from onert::exec::IFunction
virtual	~IFunction ()=default
virtual void	prepare ()

Detailed Description

Definition at line 34 of file StridedSliceLayer.h.

Constructor & Destructor Documentation

StridedSliceLayer()

onert::backend::cpu::ops::StridedSliceLayer::StridedSliceLayer

(

)

Definition at line 32 of file StridedSliceLayer.cc.

  : _input(nullptr), _begin(nullptr), _end(nullptr), _strides(nullptr), _output(nullptr),
    _begin_mask(0), _ellipsis_mask(0), _end_mask(0), _new_axis_mask(0), _shrink_axis_mask(0)
{
}

Member Function Documentation

configure()

void onert::backend::cpu::ops::StridedSliceLayer::configure	(	const IPortableTensor *	input,
		const IPortableTensor *	begin,
		const IPortableTensor *	end,
		const IPortableTensor *	strides,
		IPortableTensor *	output,
		const int32_t	begin_mask,
		const int32_t	end_mask,
		const int32_t	shrink_axis_mask
	)

Definition at line 52 of file StridedSliceLayer.cc.

{
  _input = input;
  _begin = begin;
  _end = end;
  _strides = strides;
  _output = output;
 
  _begin_mask = begin_mask;
  _ellipsis_mask = 0;
  _end_mask = end_mask;
  _new_axis_mask = 0;
  _shrink_axis_mask = shrink_axis_mask;
}

References begin.

run()

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

overridevirtual

Implements onert::exec::IFunction.

Definition at line 70 of file StridedSliceLayer.cc.

{
  if (_input->data_type() == OperandType::FLOAT32)
  {
    stridedSliceImpl<float>();
  }
  else if (_input->data_type() == OperandType::INT32)
  {
    stridedSliceImpl<int32_t>();
  }
  else if (_input->data_type() == OperandType::INT64)
  {
    stridedSliceImpl<int64_t>();
  }
  else
  {
    throw std::runtime_error{"StridedSlice: 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/StridedSliceLayer.h
• runtime/onert/backend/cpu/ops/StridedSliceLayer.cc