ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Transpose.h
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2019 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_TRANSPOSE_H__
18
#define __ONERT_IR_OPERATION_TRANSPOSE_H__
19
20
#include "ir/Operation.h"
21
22
#include <utility>
23
24
namespace
onert
25
{
26
namespace
ir
27
{
28
namespace
operation
29
{
30
31
class
Transpose
:
public
Operation
32
{
33
public
:
34
enum
Input
35
{
36
INPUT
= 0,
// for an n-D tensor, specifying the tensor to be transposed.
37
PERMUTATION
= 1,
38
};
39
40
public
:
41
Transpose
(
const
OperandIndexSequence
&inputs,
const
OperandIndexSequence
&outputs);
42
43
public
:
44
void
accept
(
OperationVisitor
&v)
const override
;
45
OpCode
opcode
() const final {
return
OpCode::Transpose; }
46
};
47
48
}
// namespace operation
49
}
// namespace ir
50
}
// namespace onert
51
52
#endif
// __ONERT_IR_OPERATION_TRANSPOSE_H__
onert::ir::OperandIndexSequence
Definition
OperandIndexSequence.h:37
onert::ir::Operation
Definition
Operation.h:34
onert::ir::operation::Transpose
Definition
Transpose.h:32
onert::ir::operation::Transpose::accept
void accept(OperationVisitor &v) const override
Definition
Transpose.cc:27
onert::ir::operation::Transpose::opcode
OpCode opcode() const final
Definition
Transpose.h:45
onert::ir::operation::Transpose::Input
Input
Definition
Transpose.h:35
onert::ir::operation::Transpose::PERMUTATION
@ PERMUTATION
Definition
Transpose.h:37
onert::ir::operation::Transpose::INPUT
@ INPUT
Definition
Transpose.h:36
onert::ir::OpCode
OpCode
Definition
OpCode.h:30
onert
Definition
CustomKernel.cc:20
onert::ir::OperationVisitor
Definition
OperationVisitor.h:28
runtime
onert
core
include
ir
operation
Transpose.h
Generated by
1.9.8