all_ops_resolver.cc

00001 /* Copyright 2018 The TensorFlow Authors. All Rights Reserved.
00002 Licensed under the Apache License, Version 2.0 (the "License");
00003 you may not use this file except in compliance with the License.
00004 You may obtain a copy of the License at
00005     http://www.apache.org/licenses/LICENSE-2.0
00006 Unless required by applicable law or agreed to in writing, software
00007 distributed under the License is distributed on an "AS IS" BASIS,
00008 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
00009 See the License for the specific language governing permissions and
00010 limitations under the License.
00011 ==============================================================================*/
00012 
00013 #include "tensorflow/lite/experimental/micro/kernels/all_ops_resolver.h"
00014 
00015 #include "tensorflow/lite/experimental/micro/kernels/micro_ops.h"
00016 
00017 namespace tflite {
00018 namespace ops {
00019 namespace micro {
00020 
00021 AllOpsResolver::AllOpsResolver() {
00022   AddBuiltin(BuiltinOperator_DEPTHWISE_CONV_2D, Register_DEPTHWISE_CONV_2D());
00023   AddBuiltin(BuiltinOperator_FULLY_CONNECTED, Register_FULLY_CONNECTED(),
00024              /* min_version */ 1,
00025              /* max_version */ 4);
00026   AddBuiltin(BuiltinOperator_MAX_POOL_2D, Register_MAX_POOL_2D());
00027   AddBuiltin(BuiltinOperator_SOFTMAX, Register_SOFTMAX());
00028   AddBuiltin(BuiltinOperator_LOGISTIC, Register_LOGISTIC());
00029   AddBuiltin(BuiltinOperator_SVDF, Register_SVDF());
00030   AddBuiltin(BuiltinOperator_CONV_2D, Register_CONV_2D());
00031   AddBuiltin(BuiltinOperator_AVERAGE_POOL_2D, Register_AVERAGE_POOL_2D());
00032   AddBuiltin(BuiltinOperator_ABS, Register_ABS());
00033   AddBuiltin(BuiltinOperator_SIN, Register_SIN());
00034   AddBuiltin(BuiltinOperator_COS, Register_COS());
00035   AddBuiltin(BuiltinOperator_LOG, Register_LOG());
00036   AddBuiltin(BuiltinOperator_SQRT, Register_SQRT());
00037   AddBuiltin(BuiltinOperator_RSQRT, Register_RSQRT());
00038   AddBuiltin(BuiltinOperator_SQUARE, Register_SQUARE());
00039   AddBuiltin(BuiltinOperator_PRELU, Register_PRELU());
00040   AddBuiltin(BuiltinOperator_FLOOR, Register_FLOOR());
00041   AddBuiltin(BuiltinOperator_MAXIMUM, Register_MAXIMUM());
00042   AddBuiltin(BuiltinOperator_MINIMUM, Register_MINIMUM());
00043   AddBuiltin(BuiltinOperator_ARG_MAX, Register_ARG_MAX());
00044   AddBuiltin(BuiltinOperator_ARG_MIN, Register_ARG_MIN());
00045   AddBuiltin(BuiltinOperator_LOGICAL_OR, Register_LOGICAL_OR());
00046   AddBuiltin(BuiltinOperator_LOGICAL_AND, Register_LOGICAL_AND());
00047   AddBuiltin(BuiltinOperator_LOGICAL_NOT, Register_LOGICAL_NOT());
00048   AddBuiltin(BuiltinOperator_RESHAPE, Register_RESHAPE());
00049   AddBuiltin(BuiltinOperator_EQUAL, Register_EQUAL());
00050   AddBuiltin(BuiltinOperator_NOT_EQUAL, Register_NOT_EQUAL());
00051   AddBuiltin(BuiltinOperator_GREATER, Register_GREATER());
00052   AddBuiltin(BuiltinOperator_GREATER_EQUAL, Register_GREATER_EQUAL());
00053   AddBuiltin(BuiltinOperator_LESS, Register_LESS());
00054   AddBuiltin(BuiltinOperator_LESS_EQUAL, Register_LESS_EQUAL());
00055   AddBuiltin(BuiltinOperator_CEIL, Register_CEIL());
00056   AddBuiltin(BuiltinOperator_ROUND, Register_ROUND());
00057   AddBuiltin(BuiltinOperator_STRIDED_SLICE, Register_STRIDED_SLICE());
00058   AddBuiltin(BuiltinOperator_PACK, Register_PACK());
00059   AddBuiltin(BuiltinOperator_SPLIT, Register_SPLIT(),
00060              /* min_version */ 1,
00061              /* max_version */ 3);
00062   AddBuiltin(BuiltinOperator_UNPACK, Register_UNPACK());
00063   AddBuiltin(BuiltinOperator_NEG, Register_NEG());
00064   AddBuiltin(BuiltinOperator_ADD, Register_ADD());
00065   AddBuiltin(BuiltinOperator_QUANTIZE, Register_QUANTIZE(), 1, 4);
00066   AddBuiltin(BuiltinOperator_DEQUANTIZE, Register_DEQUANTIZE(), 1, 4);
00067   AddBuiltin(BuiltinOperator_RELU, Register_RELU());
00068   AddBuiltin(BuiltinOperator_RELU6, Register_RELU6());
00069 }
00070 
00071 }  // namespace micro
00072 }  // namespace ops
00073 }  // namespace tflite