Join us at TensorFlow World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

hub.register_module_for_export

hub.register_module_for_export(
    module,
    export_name
)

Register a Module to be exported under export_name.

This function registers module to be exported by LatestModuleExporter under a subdirectory named export_name.

Note that export_name must be unique for each module exported from the current graph. It only controls the export subdirectory name and it has no scope effects such as the name parameter during Module instantiation.

Args:

  • module: Module instance to be exported.
  • export_name: subdirectory name to use when performing the export.

Raises:

  • ValueError: if export_name is already taken in the current graph.