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::ir::operation
21
{
22
23
void
Permute::accept
(
OperationVisitor
&v)
const
{ v.visit(*
this
); }
24
25
Permute::Permute
(
const
OperandIndex
&input,
const
OperandIndex
&output,
ir::PermuteType
type)
26
:
Operation
{
OperandConstraint
::createExact(1u)}, _type{type}
27
{
28
setInputs
({input});
29
setOutputs
({output});
30
}
31
32
}
// namespace onert::ir::operation
OperationVisitor.h
onert::ir::OperandConstraint
Definition
OperandConstraint.h:28
onert::ir::Operation
Definition
Operation.h:32
onert::ir::Operation::setInputs
void setInputs(const OperandIndexSequence &indexes)
Definition
Operation.cc:39
onert::ir::Operation::setOutputs
void setOutputs(const OperandIndexSequence &indexes)
Definition
Operation.cc:46
onert::ir::operation::Permute::Permute
Permute(const OperandIndex &input, const OperandIndex &output, PermuteType type)
Definition
Permute.cc:25
onert::ir::operation::Permute::accept
void accept(OperationVisitor &v) const override
Definition
Permute.cc:23
onert::util::Index< uint32_t, OperandIndexTag >
onert::ir::operation
Definition
AddN.h:23
onert::ir::PermuteType
PermuteType
Definition
Layout.h:36
Permute.h
onert::ir::OperationVisitor
Definition
OperationVisitor.h:26
runtime
onert
core
src
ir
operation
Permute.cc
Generated by
1.9.8