ONE - On-device Neural Engine
Loading...
Searching...
No Matches
DynamicUpdateSlice.h
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2025 Samsung Electronics Co., Ltd. All Rights Reserved
3
*
4
* Licensed under the Apache License, Version 2.0 (the "License");
5
* you may not use this file except in compliance with the License.
6
* You may obtain a copy of the License at
7
*
8
* http://www.apache.org/licenses/LICENSE-2.0
9
*
10
* Unless required by applicable law or agreed to in writing, software
11
* distributed under the License is distributed on an "AS IS" BASIS,
12
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13
* See the License for the specific language governing permissions and
14
* limitations under the License.
15
*/
16
17
#ifndef __ONERT_BACKEND_CPU_OPS_DYNAMIC_UPDATE_SLICE_LAYER_H__
18
#define __ONERT_BACKEND_CPU_OPS_DYNAMIC_UPDATE_SLICE_LAYER_H__
19
20
#include <
backend/IPortableTensor.h
>
21
22
#include <
exec/IFunction.h
>
23
#include <
ir/operation/DynamicUpdateSlice.h
>
24
25
namespace
onert::backend::cpu::ops
26
{
27
28
class
DynamicUpdateSliceLayer
:
public
::onert::exec::IFunction
29
{
30
public
:
31
DynamicUpdateSliceLayer
();
32
~DynamicUpdateSliceLayer
();
33
34
public
:
35
void
configure
(
const
IPortableTensor
*operand,
const
IPortableTensor
*update,
36
const
IPortableTensor
*indices,
IPortableTensor
*output);
37
38
void
run
()
override
;
39
40
private
:
41
const
IPortableTensor
*_operand;
42
const
IPortableTensor
*_update;
43
const
IPortableTensor
*_indices;
44
IPortableTensor
*_output;
45
};
46
47
}
// namespace onert::backend::cpu::ops
48
49
#endif
// __ONERT_BACKEND_CPU_OPS_DYNAMIC_UPDATE_SLICE_LAYER_H__
IFunction.h
IPortableTensor.h
onert::backend::IPortableTensor
A tensor class that is portable for other backends.
Definition
IPortableTensor.h:37
onert::backend::cpu::ops::DynamicUpdateSliceLayer
Definition
DynamicUpdateSlice.h:29
onert::backend::cpu::ops::DynamicUpdateSliceLayer::~DynamicUpdateSliceLayer
~DynamicUpdateSliceLayer()
onert::backend::cpu::ops::DynamicUpdateSliceLayer::DynamicUpdateSliceLayer
DynamicUpdateSliceLayer()
Definition
DynamicUpdateSlice.cc:25
onert::backend::cpu::ops::DynamicUpdateSliceLayer::configure
void configure(const IPortableTensor *operand, const IPortableTensor *update, const IPortableTensor *indices, IPortableTensor *output)
Definition
DynamicUpdateSlice.cc:33
onert::backend::cpu::ops::DynamicUpdateSliceLayer::run
void run() override
Definition
DynamicUpdateSlice.cc:48
onert::exec::IFunction
Definition
IFunction.h:24
onert::backend::cpu::ops
Definition
AddNLayer.cc:25
DynamicUpdateSlice.h
runtime
onert
backend
cpu
ops
DynamicUpdateSlice.h
Generated by
1.9.8