Google I/O returns May 18-20! Reserve space and build your schedule Register now


View source on GitHub

Options for saving to SavedModel.

This function may be used in the options argument in functions that save a SavedModel (, tf.keras.models.save_model).

namespace_whitelist List of strings containing op namespaces to whitelist when saving a model. Saving an object that uses namespaced ops must explicitly add all namespaces to the whitelist. The namespaced ops must be registered into the framework when loading the SavedModel.
save_debug_info Boolean indicating whether debug information is saved. If True, then a debug/saved_model_debug_info.pb file will be written with the contents of a GraphDebugInfo binary protocol buffer containing stack trace information for all ops and functions that are saved.

Class Variables

  • namespace_whitelist
  • save_debug_info