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::ir::operation
25
{
26
27
class
Concat
:
public
Operation
28
{
29
public
:
30
struct
Param
31
{
32
int32_t
axis
;
33
};
34
35
public
:
36
Concat
(
const
OperandIndexSequence
&inputs,
const
OperandIndexSequence
&outputs,
37
const
Param
&
param
);
38
39
public
:
40
void
accept
(
OperationVisitor
&v)
const override
;
41
OpCode
opcode
() const final {
return
OpCode::Concat; }
42
43
public
:
44
const
Param
&
param
()
const
{
return
_param; }
45
46
private
:
47
Param _param;
48
};
49
50
}
// namespace onert::ir::operation
51
52
#endif
// __ONERT_IR_OPERATION_CONCAT_H__
onert::ir::OperandIndexSequence
Definition
OperandIndexSequence.h:35
onert::ir::Operation
Definition
Operation.h:32
onert::ir::operation::Concat
Definition
Concat.h:28
onert::ir::operation::Concat::opcode
OpCode opcode() const final
Definition
Concat.h:41
onert::ir::operation::Concat::param
const Param & param() const
Definition
Concat.h:44
onert::ir::operation::Concat::accept
void accept(OperationVisitor &v) const override
Definition
Concat.cc:23
onert::ir::operation
Definition
AddN.h:23
onert::ir::OpCode
OpCode
Definition
OpCode.h:28
onert::ir::OperationVisitor
Definition
OperationVisitor.h:26
onert::ir::operation::Concat::Param
Definition
Concat.h:31
onert::ir::operation::Concat::Param::axis
int32_t axis
Definition
Concat.h:32
runtime
onert
core
include
ir
operation
Concat.h
Generated by
1.9.8