1 /* Copyright 2018 The TensorFlow Authors. All Rights Reserved.
2 Licensed under the Apache License, Version 2.0 (the "License");
3 you may not use this file except in compliance with the License.
4 You may obtain a copy of the License at
5     http://www.apache.org/licenses/LICENSE-2.0
6 Unless required by applicable law or agreed to in writing, software
7 distributed under the License is distributed on an "AS IS" BASIS,
8 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
9 See the License for the specific language governing permissions and
10 limitations under the License.
11 ==============================================================================*/
12 
13 #include "tensorflow/lite/experimental/micro/kernels/all_ops_resolver.h"
14 
15 namespace tflite {
16 namespace ops {
17 namespace micro {
18 
19 TfLiteRegistration* Register_DEPTHWISE_CONV_2D();
Micro_Register_DEPTHWISE_CONV_2D()20 TfLiteRegistration* Micro_Register_DEPTHWISE_CONV_2D() {
21   return Register_DEPTHWISE_CONV_2D();
22 }
23 
24 TfLiteRegistration* Register_FULLY_CONNECTED();
Micro_Register_FULLY_CONNECTED()25 TfLiteRegistration* Micro_Register_FULLY_CONNECTED() {
26   return Register_FULLY_CONNECTED();
27 }
28 
29 TfLiteRegistration* Register_SOFTMAX();
Micro_Register_SOFTMAX()30 TfLiteRegistration* Micro_Register_SOFTMAX() { return Register_SOFTMAX(); }
31 
AllOpsResolver()32 AllOpsResolver::AllOpsResolver() {
33   AddBuiltin(BuiltinOperator_DEPTHWISE_CONV_2D,
34              Micro_Register_DEPTHWISE_CONV_2D());
35   AddBuiltin(BuiltinOperator_FULLY_CONNECTED, Micro_Register_FULLY_CONNECTED(),
36              /* min_version */ 1,
37              /* max_version */ 2);
38   AddBuiltin(BuiltinOperator_SOFTMAX, Micro_Register_SOFTMAX());
39 }
40 
41 }  // namespace micro
42 }  // namespace ops
43 }  // namespace tflite
44