ONE - On-device Neural Engine
All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
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 27 of file SliceLayer.h.

Constructor & Destructor Documentation

◆ SliceLayer()

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

Definition at line 26 of file SliceLayer.cc.

26 : _input(nullptr), _begin(nullptr), _size(nullptr), _output(nullptr)
27{
28 // DO NOTHING
29}

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 85 of file SliceLayer.cc.

87{
88 _input = input;
89 _output = output;
90 _begin = begin;
91 _size = size;
92}
int32_t size[5]
Definition Slice.cpp:35
int32_t begin[5]
Definition Slice.cpp:33

References begin, and size.

◆ run()

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

Implements onert::exec::IFunction.

Definition at line 94 of file SliceLayer.cc.

95{
96 if (_input->data_type() == OperandType::FLOAT32)
97 {
98 sliceImpl<float>();
99 }
100 else if (_input->data_type() == OperandType::QUANT_UINT8_ASYMM)
101 {
102 sliceImpl<uint8_t>();
103 }
104 else
105 {
106 throw std::runtime_error{"Slice: unsupported data type"};
107 }
108}
ir::DataType data_type() const override final

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


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