1/* Copyright 2016 The TensorFlow Authors. All Rights Reserved.
2
3Licensed under the Apache License, Version 2.0 (the "License");
4you may not use this file except in compliance with the License.
5You may obtain a copy of the License at
6
7 http://www.apache.org/licenses/LICENSE-2.0
8
9Unless required by applicable law or agreed to in writing, software
10distributed under the License is distributed on an "AS IS" BASIS,
11WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12See the License for the specific language governing permissions and
13limitations under the License.
14==============================================================================*/
15
16#ifndef TENSORFLOW_CC_SAVED_MODEL_CONSTANTS_H_
17#define TENSORFLOW_CC_SAVED_MODEL_CONSTANTS_H_
18
19namespace tensorflow {
20
21// SavedModel assets directory.
22constexpr char kSavedModelAssetsDirectory[] = "assets";
23
24// SavedModel assets.extra directory.
25constexpr char kSavedModelAssetsExtraDirectory[] = "assets.extra";
26
27// SavedModel assets key for graph collection-def.
28constexpr char kSavedModelAssetsKey[] = "saved_model_assets";
29
30/// SavedModel legacy init op collection key. Used in v1 SavedModels.
31constexpr char kSavedModelLegacyInitOpKey[] = "legacy_init_op";
32
33/// SavedModel main op collection key. Used in v1 SavedModels.
34constexpr char kSavedModelMainOpKey[] = "saved_model_main_op";
35
36// CollectionDef key for the SavedModel train op.
37// Not exported while export_all_saved_models is experimental.
38constexpr char kSavedModelTrainOpKey[] = "saved_model_train_op";
39
40// Schema version for SavedModel.
41constexpr int kSavedModelSchemaVersion = 1;
42
43// SavedModel proto filename.
44constexpr char kSavedModelFilenamePb[] = "saved_model.pb";
45
46// SavedModel text format proto filename.
47constexpr char kSavedModelFilenamePbTxt[] = "saved_model.pbtxt";
48
49// Subdirectory where debugging related files are written.
50constexpr char kSavedModelDebugDirectory[] = "debug";
51
52// File name for GraphDebugInfo protocol buffer which corresponds to the
53// SavedModel.
54constexpr char kSavedModelDebugInfoFilenamePb[] = "saved_model_debug_info.pb";
55
56// Directory in which to save the SavedModel variables.
57constexpr char kSavedModelVariablesDirectory[] = "variables";
58
59// SavedModel variables filename.
60constexpr char kSavedModelVariablesFilename[] = "variables";
61
62// SavedModel SignatureDef keys for the initialization and train ops. Used in
63// V2 SavedModels.
64constexpr char kSavedModelInitOpSignatureKey[] = "__saved_model_init_op";
65constexpr char kSavedModelTrainOpSignatureKey[] = "__saved_model_train_op";
66
67// Key in the TensorBundle for the object graph proto.
68constexpr char kObjectGraphProtoKey[] = "_CHECKPOINTABLE_OBJECT_GRAPH";
69
70// Filename for the FingerprintDef protocol buffer.
71constexpr char kFingerprintFilenamePb[] = "fingerprint.pb";
72
73} // namespace tensorflow
74
75#endif // TENSORFLOW_CC_SAVED_MODEL_CONSTANTS_H_
76