tf.saved_model.SaveOptions

View source on GitHub

Options for saving to SavedModel.

tf.saved_model.SaveOptions(
    namespace_whitelist=None
)

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

Args:

  • 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.

Class Variables

  • namespace_whitelist