ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Shape.h
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2020 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_SHAPE_H__
18
#define __ONERT_IR_OPERATION_SHAPE_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
Shape
:
public
Operation
32
{
33
public
:
34
enum
Input
35
{
36
INPUT
= 0
37
};
38
39
public
:
40
Shape
(
const
OperandIndexSequence
&inputs,
const
OperandIndexSequence
&outputs);
41
42
public
:
43
void
accept
(
OperationVisitor
&v)
const override
;
44
OpCode
opcode
() const final {
return
OpCode::Shape; }
45
};
46
47
}
// namespace operation
48
}
// namespace ir
49
}
// namespace onert
50
51
#endif
// __ONERT_IR_OPERATION_SHAPE_H__
onert::ir::OperandIndexSequence
Definition
OperandIndexSequence.h:37
onert::ir::Operation
Definition
Operation.h:34
onert::ir::operation::Shape
Definition
Shape.h:32
onert::ir::operation::Shape::Input
Input
Definition
Shape.h:35
onert::ir::operation::Shape::INPUT
@ INPUT
Definition
Shape.h:36
onert::ir::operation::Shape::accept
void accept(OperationVisitor &v) const override
Definition
Shape.cc:27
onert::ir::operation::Shape::opcode
OpCode opcode() const final
Definition
Shape.h:44
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
Shape.h
Generated by
1.9.8