ONE - On-device Neural Engine
Loading...
Searching...
No Matches
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 54 of file SliceLayer.cc.

54 : _input(nullptr), _begin(nullptr), _size(nullptr), _output(nullptr)
55{
56 // DO NOTHING
57}

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

115{
116 _input = input;
117 _output = output;
118 _begin = begin;
119 _size = size;
120}
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 122 of file SliceLayer.cc.

123{
124 if (_input->data_type() == OperandType::FLOAT32)
125 {
126 sliceImpl<float>();
127 }
128 else if (_input->data_type() == OperandType::QUANT_UINT8_ASYMM)
129 {
130 sliceImpl<uint8_t>();
131 }
132 else
133 {
134 throw std::runtime_error{"Slice: unsupported data type"};
135 }
136}
ir::DataType data_type() const override final

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


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