ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Permute.cc
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2018 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/operation/Permute.h
"
18
#include "
ir/OperationVisitor.h
"
19
20
namespace
onert
21
{
22
namespace
ir
23
{
24
namespace
operation
25
{
26
27
void
Permute::accept
(
OperationVisitor
&v)
const
{ v.visit(*
this
); }
28
29
Permute::Permute
(
const
OperandIndex
&input,
const
OperandIndex
&output)
30
:
Operation
{
OperandConstraint
::createExact(1u)}
31
{
32
setInputs
({input});
33
setOutputs
({output});
34
}
35
36
}
// namespace operation
37
}
// namespace ir
38
}
// namespace onert
OperationVisitor.h
onert::ir::OperandConstraint
Definition
OperandConstraint.h:30
onert::ir::Operation
Definition
Operation.h:34
onert::ir::Operation::setInputs
void setInputs(const OperandIndexSequence &indexes)
Definition
Operation.cc:41
onert::ir::Operation::setOutputs
void setOutputs(const OperandIndexSequence &indexes)
Definition
Operation.cc:48
onert::ir::operation::Permute::Permute
Permute(const OperandIndex &input, const OperandIndex &output)
Definition
Permute.cc:29
onert::ir::operation::Permute::accept
void accept(OperationVisitor &v) const override
Definition
Permute.cc:27
onert::util::Index< uint32_t, OperandIndexTag >
onert
Definition
CustomKernel.cc:20
Permute.h
onert::ir::OperationVisitor
Definition
OperationVisitor.h:28
runtime
onert
core
src
ir
operation
Permute.cc
Generated by
1.9.8