Skip to content

Commit

Permalink
Internal code changes.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 535779638
  • Loading branch information
k-w-w authored and tensorflower-gardener committed May 27, 2023
1 parent 67d5c56 commit 87eea6d
Showing 1 changed file with 25 additions and 17 deletions.
42 changes: 25 additions & 17 deletions tensorflow/cc/saved_model/constants.h
Original file line number Diff line number Diff line change
Expand Up @@ -19,56 +19,64 @@ limitations under the License.
namespace tensorflow {

// SavedModel assets directory.
constexpr char kSavedModelAssetsDirectory[] = "assets";
inline constexpr char kSavedModelAssetsDirectory[] = "assets";

// SavedModel assets.extra directory.
constexpr char kSavedModelAssetsExtraDirectory[] = "assets.extra";
inline constexpr char kSavedModelAssetsExtraDirectory[] = "assets.extra";

// SavedModel assets key for graph collection-def.
constexpr char kSavedModelAssetsKey[] = "saved_model_assets";
inline constexpr char kSavedModelAssetsKey[] = "saved_model_assets";

/// SavedModel legacy init op collection key. Used in v1 SavedModels.
constexpr char kSavedModelLegacyInitOpKey[] = "legacy_init_op";
inline constexpr char kSavedModelLegacyInitOpKey[] = "legacy_init_op";

/// SavedModel main op collection key. Used in v1 SavedModels.
constexpr char kSavedModelMainOpKey[] = "saved_model_main_op";
inline constexpr char kSavedModelMainOpKey[] = "saved_model_main_op";

// CollectionDef key for the SavedModel train op.
// Not exported while export_all_saved_models is experimental.
constexpr char kSavedModelTrainOpKey[] = "saved_model_train_op";
inline constexpr char kSavedModelTrainOpKey[] = "saved_model_train_op";

// Schema version for SavedModel.
constexpr int kSavedModelSchemaVersion = 1;
inline constexpr int kSavedModelSchemaVersion = 1;

// SavedModel proto filename prefix.
inline constexpr char kSavedModelFilenamePrefix[] = "saved_model";

// SavedModel proto filename.
constexpr char kSavedModelFilenamePb[] = "saved_model.pb";
inline constexpr char kSavedModelFilenamePb[] = "saved_model.pb";

// SavedModel chunked proto filename.
inline constexpr char kSavedModelFilenameCpb[] = "saved_model.cpb";

// SavedModel text format proto filename.
constexpr char kSavedModelFilenamePbTxt[] = "saved_model.pbtxt";
inline constexpr char kSavedModelFilenamePbTxt[] = "saved_model.pbtxt";

// Subdirectory where debugging related files are written.
constexpr char kSavedModelDebugDirectory[] = "debug";
inline constexpr char kSavedModelDebugDirectory[] = "debug";

// File name for GraphDebugInfo protocol buffer which corresponds to the
// SavedModel.
constexpr char kSavedModelDebugInfoFilenamePb[] = "saved_model_debug_info.pb";
inline constexpr char kSavedModelDebugInfoFilenamePb[] =
"saved_model_debug_info.pb";

// Directory in which to save the SavedModel variables.
constexpr char kSavedModelVariablesDirectory[] = "variables";
inline constexpr char kSavedModelVariablesDirectory[] = "variables";

// SavedModel variables filename.
constexpr char kSavedModelVariablesFilename[] = "variables";
inline constexpr char kSavedModelVariablesFilename[] = "variables";

// SavedModel SignatureDef keys for the initialization and train ops. Used in
// V2 SavedModels.
constexpr char kSavedModelInitOpSignatureKey[] = "__saved_model_init_op";
constexpr char kSavedModelTrainOpSignatureKey[] = "__saved_model_train_op";
inline constexpr char kSavedModelInitOpSignatureKey[] = "__saved_model_init_op";
inline constexpr char kSavedModelTrainOpSignatureKey[] =
"__saved_model_train_op";

// Key in the TensorBundle for the object graph proto.
constexpr char kObjectGraphProtoKey[] = "_CHECKPOINTABLE_OBJECT_GRAPH";
inline constexpr char kObjectGraphProtoKey[] = "_CHECKPOINTABLE_OBJECT_GRAPH";

// Filename for the FingerprintDef protocol buffer.
constexpr char kFingerprintFilenamePb[] = "fingerprint.pb";
inline constexpr char kFingerprintFilenamePb[] = "fingerprint.pb";

} // namespace tensorflow

Expand Down

0 comments on commit 87eea6d

Please sign in to comment.