forked from OAID/Tengine
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathop_name.h
129 lines (126 loc) · 5.76 KB
/
op_name.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* License); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* AS IS BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
/*
* Copyright (c) 2021, OPEN AI LAB
* Author: [email protected]
*/
#pragma once
#define OP_GENERIC_NAME "Generic"
#define OP_ABSVAL_NAME "Absval"
#define OP_ADD_N_NAME "Add_n"
#define OP_ARGMAX_NAME "ArgMax"
#define OP_ARGMIN_NAME "ArgMin"
#define OP_BATCHNORM_NAME "BatchNormalize"
#define OP_BATCHTOSPACEND_NAME "Batchtospacend"
#define OP_BIAS_NAME "Bias"
#define OP_BROADMUL_NAME "BroadMul"
#define OP_CAST_NAME "Cast"
#define OP_CEIL_NAME "Ceil"
#define OP_CLIP_NAME "Clip"
#define OP_COMPARISON_NAME "Comparison"
#define OP_CONCAT_NAME "Concat"
#define OP_CONV_NAME "Convolution"
#define OP_CONST_NAME "Const"
#define OP_CROP_NAME "Crop"
#define OP_DECONV_NAME "Deconvolution"
#define OP_DEPTHTOSPACE_NAME "Depthtospace"
#define OP_DETECTION_OUTPUT_NAME "DetectionOutput"
#define OP_DETECTION_POSTPROCESS_NAME "DetectionPostProcess"
#define OP_DROPOUT_NAME "Dropout"
#define OP_ELTWISE_NAME "Eltwise"
#define OP_ELU_NAME "Elu"
#define OP_EMBEDDING_NAME "Embedding"
#define OP_EXPANDDIMS_NAME "Expanddims"
#define OP_FC_NAME "FullyConnected"
#define OP_FLATTEN_NAME "Flatten"
#define OP_GATHER_NAME "Gather"
#define OP_GEMM_NAME "Gemm"
#define OP_GRU_NAME "Gru"
#define OP_HARDSIGMOID_NAME "HardSigmoid"
#define OP_HARDSWISH_NAME "Hardswish"
#define OP_INPUT_NAME "InputOp"
#define OP_INSTANCENORM_NAME "InstanceNorm"
#define OP_INTERP_NAME "Interp"
#define OP_LOGICAL_NAME "Logical"
#define OP_LOGISTIC_NAME "Logistic"
#define OP_LRN_NAME "Lrn"
#define OP_LSTM_NAME "Lstm"
#define OP_MATMUL_NAME "Matmul"
#define OP_MAXIMUM_NAME "Maximum"
#define OP_MEAN_NAME "Mean"
#define OP_MINIMUM_NAME "Minimum"
#define OP_MVN_NAME "Mvn"
#define OP_NOOP_NAME "Noop"
#define OP_NORMALIZE_NAME "Normalize"
#define OP_PAD_NAME "Pad"
#define OP_PERMUTE_NAME "Permute"
#define OP_POOL_NAME "Pooling"
#define OP_PRELU_NAME "PReLU"
#define OP_PRIORBOX_NAME "PriorBox"
#define OP_PSROIPOOLING_NAME "Psroipooling"
#define OP_REDUCEL2_NAME "ReduceL2"
#define OP_REDUCTION_NAME "Reduction"
#define OP_REGION_NAME "Region"
#define OP_RELU_NAME "ReLU"
#define OP_RELU6_NAME "ReLU6"
#define OP_REORG_NAME "Reorg"
#define OP_RESHAPE_NAME "Reshape"
#define OP_RESIZE_NAME "Resize"
#define OP_REVERSE_NAME "Reverse"
#define OP_RNN_NAME "RNN"
#define OP_ROIALIGN_NAME "Roialign"
#define OP_ROIPOOLING_NAME "RoiPooling"
#define OP_ROUND_NAME "Round"
#define OP_RPN_NAME "Rpn"
#define OP_SCALE_NAME "Scale"
#define OP_SELU_NAME "Selu"
#define OP_SHUFFLECHANNEL_NAME "ShuffleChannel"
#define OP_SIGMOID_NAME "Sigmoid"
#define OP_SLICE_NAME "Slice"
#define OP_SOFTMAX_NAME "Softmax"
#define OP_SPACETOBATCHND_NAME "Spacetobatchnd"
#define OP_SPACETODEPTH_NAME "Spacetodepth"
#define OP_SPARSETODENSE_NAME "SparseToDense"
#define OP_SPLIT_NAME "Split"
#define OP_SQUAREDDIFFERENCE_NAME "SquaredDifference"
#define OP_SQUEEZE_NAME "Squeeze"
#define OP_STRIDEDSLICE_NAME "StridedSlice"
#define OP_SWAP_AXIS_NAME "SwapAxis"
#define OP_TANH_NAME "Tanh"
#define OP_THRESHOLD_NAME "Threshold"
#define OP_TOPKV2_NAME "Topkv2"
#define OP_TRANSPOSE_NAME "Transpose"
#define OP_UNARY_NAME "Unary"
#define OP_UNSQUEEZE_NAME "Unsqueeze"
#define OP_UPSAMPLE_NAME "Upsample"
#define OP_ZEROSLIKE_NAME "ZerosLike"
#define OP_MISH_NAME "Mish"
#define OP_LOGSOFTMAX_NAME "LogSoftmax"
#define OP_RELU1_NAME "ReLU1"
#define OP_L2NORMALIZATION_NAME "L2Normalization"
#define OP_L2POOL_NAME "L2Pool"
#define OP_TILE_NAME "Tile"
#define OP_SHAPE_NAME "Shape"
#define OP_SCATTER_NAME "Scatter"
#define OP_WHERE_NAME "Where"
#define OP_SOFTPLUS_NAME "Softplus"
#define OP_RECIPROCAL_NAME "Reciprocal"
#define OP_SPATIALTRANSFORMER_NAME "SpatialTransformer"
#define OP_EXPAND_NAME "Expand"
#define OP_GELU_NAME "Gelu"