ONE - On-device Neural Engine
Loading...
Searching...
No Matches
ElementwiseActivation.cc
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2023 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
#include "
ir/train/operation/ElementwiseActivation.h
"
18
19
#include "
ir/OperationVisitor.h
"
20
#include "
ir/train/TrainableOperationVisitor.h
"
21
22
namespace
onert
23
{
24
namespace
ir
25
{
26
namespace
train
27
{
28
namespace
operation
29
{
30
31
std::unique_ptr<ITrainableOperation>
ElementwiseActivation::clone
()
const
32
{
33
return
std::make_unique<ElementwiseActivation>(*
this
);
34
}
35
36
void
ElementwiseActivation::accept
(
OperationVisitor
&v)
const
{ v.visit(*
this
); }
37
38
void
ElementwiseActivation::accept
(
TrainableOperationVisitor
&v)
const
{ v.visit(*
this
); }
39
40
ElementwiseActivation::ElementwiseActivation
(
const
OperationType
&operation)
41
:
OperationType
{operation.getInputs(), operation.getOutputs(), operation.param()}
42
{
43
// DO NOTHING
44
}
45
46
}
// namespace operation
47
}
// namespace train
48
}
// namespace ir
49
}
// namespace onert
OperationVisitor.h
TrainableOperationVisitor.h
onert::ir::operation::ElementwiseActivation
Definition
ElementwiseActivation.h:30
onert::ir::train::operation::ElementwiseActivation::accept
void accept(OperationVisitor &v) const override
Definition
ElementwiseActivation.cc:36
onert::ir::train::operation::ElementwiseActivation::ElementwiseActivation
ElementwiseActivation(const OperationType &operation)
Definition
ElementwiseActivation.cc:40
onert::ir::train::operation::ElementwiseActivation::clone
std::unique_ptr< ITrainableOperation > clone() const override
Definition
ElementwiseActivation.cc:31
onert
Definition
CustomKernel.cc:20
onert::ir::OperationVisitor
Definition
OperationVisitor.h:28
onert::ir::train::TrainableOperationVisitor
Definition
TrainableOperationVisitor.h:30
ElementwiseActivation.h
runtime
onert
core
src
ir
train
operation
ElementwiseActivation.cc
Generated by
1.9.8