1 | /* Copyright 2019 The TensorFlow Authors. All Rights Reserved. |
---|---|
2 | |
3 | Licensed under the Apache License, Version 2.0 (the "License"); |
4 | you may not use this file except in compliance with the License. |
5 | You may obtain a copy of the License at |
6 | |
7 | http://www.apache.org/licenses/LICENSE-2.0 |
8 | |
9 | Unless required by applicable law or agreed to in writing, software |
10 | distributed under the License is distributed on an "AS IS" BASIS, |
11 | WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
12 | See the License for the specific language governing permissions and |
13 | limitations under the License. |
14 | ==============================================================================*/ |
15 | #ifndef TENSORFLOW_LITE_KERNELS_BUILTIN_OP_KERNELS_H_ |
16 | #define TENSORFLOW_LITE_KERNELS_BUILTIN_OP_KERNELS_H_ |
17 | |
18 | #include "tensorflow/lite/c/common.h" |
19 | |
20 | namespace tflite { |
21 | namespace ops { |
22 | namespace builtin { |
23 | |
24 | // Forward declaration of all builtin op kernel registration methods. These |
25 | // registrations are included with the standard `BuiltinOpResolver`. |
26 | // |
27 | // This header is particularly useful in cases where only a subset of ops are |
28 | // needed. In such cases, the client can selectively add only the registrations |
29 | // their model requires, using a custom `OpResolver` or `MutableOpResolver`. |
30 | // Selective registration in turn allows the linker to strip unused kernels. |
31 | // |
32 | // TODO(b/184734878): auto-generate this header file from the BuiltinOperator |
33 | // enum in the FlatBuffer schema. |
34 | |
35 | TfLiteRegistration* Register_ABS(); |
36 | TfLiteRegistration* Register_ADD(); |
37 | TfLiteRegistration* Register_ADD_N(); |
38 | TfLiteRegistration* Register_ARG_MAX(); |
39 | TfLiteRegistration* Register_ARG_MIN(); |
40 | TfLiteRegistration* Register_ASSIGN_VARIABLE(); |
41 | TfLiteRegistration* Register_ATAN2(); |
42 | TfLiteRegistration* Register_AVERAGE_POOL_2D(); |
43 | TfLiteRegistration* Register_BATCH_TO_SPACE_ND(); |
44 | TfLiteRegistration* Register_BATCH_MATMUL(); |
45 | TfLiteRegistration* Register_BIDIRECTIONAL_SEQUENCE_LSTM(); |
46 | TfLiteRegistration* Register_BIDIRECTIONAL_SEQUENCE_RNN(); |
47 | TfLiteRegistration* Register_BROADCAST_ARGS(); |
48 | TfLiteRegistration* Register_BROADCAST_TO(); |
49 | TfLiteRegistration* Register_BUCKETIZE(); |
50 | TfLiteRegistration* Register_CALL_ONCE(); |
51 | TfLiteRegistration* Register_CAST(); |
52 | TfLiteRegistration* Register_CEIL(); |
53 | TfLiteRegistration* Register_COMPLEX_ABS(); |
54 | TfLiteRegistration* Register_CONCATENATION(); |
55 | TfLiteRegistration* Register_CONV_2D(); |
56 | TfLiteRegistration* Register_CONV_3D(); |
57 | TfLiteRegistration* Register_CONV_3D_TRANSPOSE(); |
58 | TfLiteRegistration* Register_COS(); |
59 | TfLiteRegistration* Register_CUMSUM(); |
60 | TfLiteRegistration* Register_DENSIFY(); |
61 | TfLiteRegistration* Register_DEPTH_TO_SPACE(); |
62 | TfLiteRegistration* Register_DEPTHWISE_CONV_2D(); |
63 | TfLiteRegistration* Register_DEQUANTIZE(); |
64 | TfLiteRegistration* Register_DIV(); |
65 | TfLiteRegistration* Register_DYNAMIC_UPDATE_SLICE(); |
66 | TfLiteRegistration* Register_ELU(); |
67 | TfLiteRegistration* Register_EMBEDDING_LOOKUP(); |
68 | TfLiteRegistration* Register_EMBEDDING_LOOKUP_SPARSE(); |
69 | TfLiteRegistration* Register_EQUAL(); |
70 | TfLiteRegistration* Register_EXP(); |
71 | TfLiteRegistration* Register_EXPAND_DIMS(); |
72 | TfLiteRegistration* Register_FAKE_QUANT(); |
73 | TfLiteRegistration* Register_FILL(); |
74 | TfLiteRegistration* Register_FLOOR(); |
75 | TfLiteRegistration* Register_FLOOR_DIV(); |
76 | TfLiteRegistration* Register_FLOOR_MOD(); |
77 | TfLiteRegistration* Register_FULLY_CONNECTED(); |
78 | TfLiteRegistration* Register_GATHER(); |
79 | TfLiteRegistration* Register_GATHER_ND(); |
80 | TfLiteRegistration* Register_GELU(); |
81 | TfLiteRegistration* Register_GREATER(); |
82 | TfLiteRegistration* Register_GREATER_EQUAL(); |
83 | TfLiteRegistration* Register_HARD_SWISH(); |
84 | TfLiteRegistration* Register_HASHTABLE(); |
85 | TfLiteRegistration* Register_HASHTABLE_FIND(); |
86 | TfLiteRegistration* Register_HASHTABLE_LOOKUP(); |
87 | TfLiteRegistration* Register_HASHTABLE_IMPORT(); |
88 | TfLiteRegistration* Register_HASHTABLE_SIZE(); |
89 | TfLiteRegistration* Register_IF(); |
90 | TfLiteRegistration* Register_IMAG(); |
91 | TfLiteRegistration* Register_L2_NORMALIZATION(); |
92 | TfLiteRegistration* Register_L2_POOL_2D(); |
93 | TfLiteRegistration* Register_LEAKY_RELU(); |
94 | TfLiteRegistration* Register_LESS(); |
95 | TfLiteRegistration* Register_LESS_EQUAL(); |
96 | TfLiteRegistration* Register_LOCAL_RESPONSE_NORMALIZATION(); |
97 | TfLiteRegistration* Register_LOG(); |
98 | TfLiteRegistration* Register_LOGICAL_AND(); |
99 | TfLiteRegistration* Register_LOGICAL_NOT(); |
100 | TfLiteRegistration* Register_LOGICAL_OR(); |
101 | TfLiteRegistration* Register_LOGISTIC(); |
102 | TfLiteRegistration* Register_LOG_SOFTMAX(); |
103 | TfLiteRegistration* Register_LSH_PROJECTION(); |
104 | TfLiteRegistration* Register_LSTM(); |
105 | TfLiteRegistration* Register_MATRIX_DIAG(); |
106 | TfLiteRegistration* Register_MATRIX_SET_DIAG(); |
107 | TfLiteRegistration* Register_MAXIMUM(); |
108 | TfLiteRegistration* Register_MAX_POOL_2D(); |
109 | TfLiteRegistration* Register_MEAN(); |
110 | TfLiteRegistration* Register_MINIMUM(); |
111 | TfLiteRegistration* Register_MIRROR_PAD(); |
112 | TfLiteRegistration* Register_MUL(); |
113 | TfLiteRegistration* Register_NEG(); |
114 | TfLiteRegistration* Register_NON_MAX_SUPPRESSION_V4(); |
115 | TfLiteRegistration* Register_NON_MAX_SUPPRESSION_V5(); |
116 | TfLiteRegistration* Register_NOT_EQUAL(); |
117 | TfLiteRegistration* Register_ONE_HOT(); |
118 | TfLiteRegistration* Register_PACK(); |
119 | TfLiteRegistration* Register_PAD(); |
120 | TfLiteRegistration* Register_PADV2(); |
121 | TfLiteRegistration* Register_POW(); |
122 | TfLiteRegistration* Register_PRELU(); |
123 | TfLiteRegistration* Register_QUANTIZE(); |
124 | TfLiteRegistration* Register_MULTINOMIAL(); |
125 | TfLiteRegistration* Register_RANDOM_STANDARD_NORMAL(); |
126 | TfLiteRegistration* Register_RANDOM_UNIFORM(); |
127 | TfLiteRegistration* Register_RANGE(); |
128 | TfLiteRegistration* Register_RANK(); |
129 | TfLiteRegistration* Register_READ_VARIABLE(); |
130 | TfLiteRegistration* Register_REAL(); |
131 | TfLiteRegistration* Register_REDUCE_ALL(); |
132 | TfLiteRegistration* Register_REDUCE_ANY(); |
133 | TfLiteRegistration* Register_REDUCE_MAX(); |
134 | TfLiteRegistration* Register_REDUCE_MIN(); |
135 | TfLiteRegistration* Register_REDUCE_PROD(); |
136 | TfLiteRegistration* Register_RELU(); |
137 | TfLiteRegistration* Register_RELU6(); |
138 | TfLiteRegistration* Register_RELU_N1_TO_1(); |
139 | TfLiteRegistration* Register_RELU_0_TO_1(); |
140 | TfLiteRegistration* Register_RESHAPE(); |
141 | TfLiteRegistration* Register_RESIZE_BILINEAR(); |
142 | TfLiteRegistration* Register_RESIZE_NEAREST_NEIGHBOR(); |
143 | TfLiteRegistration* Register_REVERSE_SEQUENCE(); |
144 | TfLiteRegistration* Register_REVERSE_V2(); |
145 | TfLiteRegistration* Register_RFFT2D(); |
146 | TfLiteRegistration* Register_RNN(); |
147 | TfLiteRegistration* Register_ROUND(); |
148 | TfLiteRegistration* Register_RSQRT(); |
149 | TfLiteRegistration* Register_SCATTER_ND(); |
150 | TfLiteRegistration* Register_SEGMENT_SUM(); |
151 | TfLiteRegistration* Register_SELECT(); |
152 | TfLiteRegistration* Register_SELECT_V2(); |
153 | TfLiteRegistration* Register_SHAPE(); |
154 | TfLiteRegistration* Register_SIGN(); |
155 | TfLiteRegistration* Register_SIN(); |
156 | TfLiteRegistration* Register_SKIP_GRAM(); |
157 | TfLiteRegistration* Register_SLICE(); |
158 | TfLiteRegistration* Register_SOFTMAX(); |
159 | TfLiteRegistration* Register_SPACE_TO_BATCH_ND(); |
160 | TfLiteRegistration* Register_SPACE_TO_DEPTH(); |
161 | TfLiteRegistration* Register_SPARSE_TO_DENSE(); |
162 | TfLiteRegistration* Register_SPLIT(); |
163 | TfLiteRegistration* Register_SPLIT_V(); |
164 | TfLiteRegistration* Register_SQRT(); |
165 | TfLiteRegistration* Register_SQUARE(); |
166 | TfLiteRegistration* Register_SQUARED_DIFFERENCE(); |
167 | TfLiteRegistration* Register_SQUEEZE(); |
168 | TfLiteRegistration* Register_STRIDED_SLICE(); |
169 | TfLiteRegistration* Register_SUB(); |
170 | TfLiteRegistration* Register_SUM(); |
171 | TfLiteRegistration* Register_SVDF(); |
172 | TfLiteRegistration* Register_TANH(); |
173 | TfLiteRegistration* Register_TILE(); |
174 | TfLiteRegistration* Register_TOPK_V2(); |
175 | TfLiteRegistration* Register_TRANSPOSE(); |
176 | TfLiteRegistration* Register_TRANSPOSE_CONV(); |
177 | TfLiteRegistration* Register_UNIDIRECTIONAL_SEQUENCE_LSTM(); |
178 | TfLiteRegistration* Register_UNIDIRECTIONAL_SEQUENCE_RNN(); |
179 | TfLiteRegistration* Register_UNIQUE(); |
180 | TfLiteRegistration* Register_UNPACK(); |
181 | TfLiteRegistration* Register_UNSORTED_SEGMENT_MAX(); |
182 | TfLiteRegistration* Register_UNSORTED_SEGMENT_MIN(); |
183 | TfLiteRegistration* Register_UNSORTED_SEGMENT_PROD(); |
184 | TfLiteRegistration* Register_UNSORTED_SEGMENT_SUM(); |
185 | TfLiteRegistration* Register_VAR_HANDLE(); |
186 | TfLiteRegistration* Register_WHERE(); |
187 | TfLiteRegistration* Register_WHILE(); |
188 | TfLiteRegistration* Register_ZEROS_LIKE(); |
189 | |
190 | } // namespace builtin |
191 | } // namespace ops |
192 | } // namespace tflite |
193 | |
194 | #endif // TENSORFLOW_LITE_KERNELS_BUILTIN_OP_KERNELS_H_ |
195 |