1 /* Copyright 2018 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_EXPERIMENTAL_WRITER_ENUM_MAPPING_H_ 16 #define TENSORFLOW_LITE_EXPERIMENTAL_WRITER_ENUM_MAPPING_H_ 17 18 #include "tensorflow/lite/builtin_op_data.h" 19 #include "tensorflow/lite/schema/reflection/schema_generated.h" 20 21 // TODO(aselle): Ideally extract this from the schema. 22 23 namespace tflite { 24 TfLiteActivationToSchemaActivation(TfLiteFusedActivation act)25inline ActivationFunctionType TfLiteActivationToSchemaActivation( 26 TfLiteFusedActivation act) { 27 switch (act) { 28 case kTfLiteActNone: 29 return ActivationFunctionType_NONE; 30 case kTfLiteActRelu: 31 return ActivationFunctionType_RELU; 32 case kTfLiteActRelu1: 33 return ActivationFunctionType_RELU_N1_TO_1; 34 case kTfLiteActRelu6: 35 return ActivationFunctionType_RELU6; 36 case kTfLiteActTanh: 37 return ActivationFunctionType_TANH; 38 case kTfLiteActSignBit: 39 return ActivationFunctionType_SIGN_BIT; 40 case kTfLiteActSigmoid: 41 return ActivationFunctionType_NONE; // TODO(aselle): Add to schema 42 } 43 return ActivationFunctionType_NONE; 44 } 45 TfLitePaddingToSchemaPadding(TfLitePadding padding)46inline Padding TfLitePaddingToSchemaPadding(TfLitePadding padding) { 47 switch (padding) { 48 case kTfLitePaddingUnknown: 49 return Padding_SAME; // TODO(aselle): Consider an error. 50 case kTfLitePaddingSame: 51 return Padding_SAME; 52 case kTfLitePaddingValid: 53 return Padding_VALID; 54 } 55 return Padding_SAME; // TODO(aselle): Consider an error. 56 } 57 TfLiteTypeToSchemaType(TfLiteType type)58inline TensorType TfLiteTypeToSchemaType(TfLiteType type) { 59 switch (type) { 60 // case kTfLiteNoType: return TensorType_NONE; 61 case kTfLiteNoType: 62 return TensorType_FLOAT32; // TODO(aselle): Consider an error. 63 case kTfLiteFloat32: 64 return TensorType_FLOAT32; 65 case kTfLiteInt32: 66 return TensorType_INT32; 67 case kTfLiteUInt8: 68 return TensorType_UINT8; 69 case kTfLiteInt64: 70 return TensorType_INT64; 71 case kTfLiteString: 72 return TensorType_STRING; 73 case kTfLiteBool: 74 return TensorType_BOOL; 75 case kTfLiteInt16: 76 return TensorType_INT16; 77 case kTfLiteComplex64: 78 return TensorType_COMPLEX64; 79 } 80 // TODO(aselle): consider an error 81 } 82 83 inline FullyConnectedOptionsWeightsFormat FullyConnectedOptionsWeightsFormatToSchema(TfLiteFullyConnectedWeightsFormat format)84FullyConnectedOptionsWeightsFormatToSchema( 85 TfLiteFullyConnectedWeightsFormat format) { 86 switch (format) { 87 case kTfLiteFullyConnectedWeightsFormatDefault: 88 return FullyConnectedOptionsWeightsFormat_DEFAULT; 89 case kTfLiteFullyConnectedWeightsFormatShuffled4x16Int8: 90 return FullyConnectedOptionsWeightsFormat_SHUFFLED4x16INT8; 91 } 92 } 93 LSTMKernelTypeToSchema(TfLiteLSTMKernelType type)94inline LSTMKernelType LSTMKernelTypeToSchema(TfLiteLSTMKernelType type) { 95 switch (type) { 96 case kTfLiteLSTMFullKernel: 97 return LSTMKernelType_FULL; 98 case kTfLiteLSTMBasicKernel: 99 return LSTMKernelType_BASIC; 100 } 101 } 102 LSHProjectionTypeToSchema(TfLiteLSHProjectionType type)103inline LSHProjectionType LSHProjectionTypeToSchema( 104 TfLiteLSHProjectionType type) { 105 switch (type) { 106 case kTfLiteLshProjectionUnknown: 107 return LSHProjectionType_UNKNOWN; 108 case kTfLiteLshProjectionSparse: 109 return LSHProjectionType_SPARSE; 110 case kTfLiteLshProjectionDense: 111 return LSHProjectionType_DENSE; 112 } 113 } 114 MirrorPaddingModeToSchema(TfLiteMirrorPaddingMode mode)115inline MirrorPadMode MirrorPaddingModeToSchema(TfLiteMirrorPaddingMode mode) { 116 switch (mode) { 117 case kTfLiteMirrorPaddingUnknown: 118 return MirrorPadMode_REFLECT; // TODO(aselle): consider an error 119 case kTfLiteMirrorPaddingReflect: 120 return MirrorPadMode_REFLECT; 121 case kTfLiteMirrorPaddingSymmetric: 122 return MirrorPadMode_SYMMETRIC; 123 } 124 } 125 CombinerTypeToSchema(TfLiteCombinerType type)126inline CombinerType CombinerTypeToSchema(TfLiteCombinerType type) { 127 switch (type) { 128 case kTfLiteCombinerTypeSum: 129 return CombinerType_SUM; 130 case kTfLiteCombinerTypeMean: 131 return CombinerType_MEAN; 132 case kTfLiteCombinerTypeSqrtn: 133 return CombinerType_SQRTN; 134 } 135 } 136 137 // int 138 139 } // namespace tflite 140 #endif // TENSORFLOW_LITE_EXPERIMENTAL_WRITER_ENUM_MAPPING_H_ 141