ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Concat.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_CONCAT_H__
18
#define __ONERT_IR_OPERATION_CONCAT_H__
19
20
#include <memory>
21
22
#include "ir/Operation.h"
23
24
namespace
onert
25
{
26
namespace
ir
27
{
28
namespace
operation
29
{
30
31
class
Concat
:
public
Operation
32
{
33
public
:
34
struct
Param
35
{
36
int32_t
axis
;
37
};
38
39
public
:
40
Concat
(
const
OperandIndexSequence
&inputs,
const
OperandIndexSequence
&outputs,
41
const
Param
&
param
);
42
43
public
:
44
void
accept
(
OperationVisitor
&v)
const override
;
45
OpCode
opcode
() const final {
return
OpCode::Concat; }
46
47
public
:
48
const
Param
&
param
()
const
{
return
_param; }
49
50
private
:
51
Param _param;
52
};
53
54
}
// namespace operation
55
}
// namespace ir
56
}
// namespace onert
57
58
#endif
// __ONERT_IR_OPERATION_CONCAT_H__
onert::ir::OperandIndexSequence
Definition
OperandIndexSequence.h:37
onert::ir::Operation
Definition
Operation.h:34
onert::ir::operation::Concat
Definition
Concat.h:32
onert::ir::operation::Concat::opcode
OpCode opcode() const final
Definition
Concat.h:45
onert::ir::operation::Concat::param
const Param & param() const
Definition
Concat.h:48
onert::ir::operation::Concat::accept
void accept(OperationVisitor &v) const override
Definition
Concat.cc:27
onert::ir::OpCode
OpCode
Definition
OpCode.h:30
onert
Definition
CustomKernel.cc:20
onert::ir::OperationVisitor
Definition
OperationVisitor.h:28
onert::ir::operation::Concat::Param
Definition
Concat.h:35
onert::ir::operation::Concat::Param::axis
int32_t axis
Definition
Concat.h:36
runtime
onert
core
include
ir
operation
Concat.h
Generated by
1.9.8