1 /* Copyright 2016 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 16 #ifndef TENSORFLOW_CC_SAVED_MODEL_CONSTANTS_H_ 17 #define TENSORFLOW_CC_SAVED_MODEL_CONSTANTS_H_ 18 19 namespace tensorflow { 20 21 /// SavedModel assets directory. 22 constexpr char kSavedModelAssetsDirectory[] = "assets"; 23 24 /// SavedModel assets.extra directory. 25 constexpr char kSavedModelAssetsExtraDirectory[] = "assets.extra"; 26 27 /// SavedModel assets key for graph collection-def. 28 constexpr char kSavedModelAssetsKey[] = "saved_model_assets"; 29 30 /// SavedModel proto filename. 31 constexpr char kSavedModelFilenamePb[] = "saved_model.pb"; 32 33 /// SavedModel text format proto filename. 34 constexpr char kSavedModelFilenamePbTxt[] = "saved_model.pbtxt"; 35 36 /// SavedModel legacy init op collection key. Used in v1 SavedModels. 37 constexpr char kSavedModelLegacyInitOpKey[] = "legacy_init_op"; 38 39 /// SavedModel main op collection key. Used in v1 SavedModels. 40 constexpr char kSavedModelMainOpKey[] = "saved_model_main_op"; 41 42 /// Directory in which to save the SavedModel variables. 43 constexpr char kSavedModelVariablesDirectory[] = "variables"; 44 45 /// SavedModel variables filename. 46 constexpr char kSavedModelVariablesFilename[] = "variables"; 47 48 /// SavedModel SignatureDef keys for the initialization and train ops. Used in 49 /// V2 SavedModels. 50 constexpr char kSavedModelInitOpSignatureKey[] = "__saved_model_init_op"; 51 constexpr char kSavedModelTrainOpSignatureKey[] = "__saved_model_train_op"; 52 53 // Key in the TensorBundle for the object graph proto. 54 constexpr char kObjectGraphProtoKey[] = "_CHECKPOINTABLE_OBJECT_GRAPH"; 55 56 } // namespace tensorflow 57 58 #endif // TENSORFLOW_CC_SAVED_MODEL_CONSTANTS_H_ 59