ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Permute.h
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
#ifndef __ONERT_IR_OPERATION_PERMUTE_H__
18
#define __ONERT_IR_OPERATION_PERMUTE_H__
19
20
#include "ir/Operation.h"
21
22
namespace
onert::backend
23
{
24
class
BackendContext
;
25
}
// namespace onert::backend
26
27
namespace
onert::ir::operation
28
{
29
49
class
Permute
:
public
Operation
50
{
51
public
:
52
void
accept
(
OperationVisitor
&v)
const override
;
53
OpCode
opcode
() const final {
return
OpCode::Permute; }
54
55
public
:
56
Permute
(
const
OperandIndex
&input,
const
OperandIndex
&output,
PermuteType
type);
57
58
public
:
59
PermuteType
getPermuteType
()
const
{
return
_type; }
60
61
private
:
62
PermuteType
_type =
PermuteType::SAME
;
63
};
64
65
}
// namespace onert::ir::operation
66
67
#endif
// __ONERT_IR_OPERATION_PERMUTE_H__
onert::ir::Operation
Definition
Operation.h:32
onert::ir::operation::Permute
Class to represent Permute operation.
Definition
Permute.h:50
onert::ir::operation::Permute::accept
void accept(OperationVisitor &v) const override
Definition
Permute.cc:23
onert::ir::operation::Permute::getPermuteType
PermuteType getPermuteType() const
Definition
Permute.h:59
onert::ir::operation::Permute::opcode
OpCode opcode() const final
Definition
Permute.h:53
onert::util::Index< uint32_t, OperandIndexTag >
onert::backend::acl_cl::BackendContext
acl_common::AclBackendContext< TensorBuilder, ConstantInitializer, KernelGenerator, Optimizer > BackendContext
Definition
BackendContext.h:32
onert::backend
Definition
Backend.h:32
onert::ir::operation
Definition
AddN.h:23
onert::ir::OpCode
OpCode
Definition
OpCode.h:28
onert::ir::PermuteType
PermuteType
Definition
Layout.h:36
onert::ir::PermuteType::SAME
@ SAME
onert::ir::OperationVisitor
Definition
OperationVisitor.h:26
runtime
onert
core
include
ir
operation
Permute.h
Generated by
1.9.8