Function export_all

tensorflow/python/distribute/hvd_strategy.py:659–755

r'''Build a SavedModel from variables in checkpoint. Args: export_dir_base: A string containing a directory to write the exported graph and checkpoints. checkpoint_path: A path to a checkpoint. signature_defs_and_main_op_fn: Function returns signa

(
            export_dir_base,
            checkpoint_path,
            signature_defs_and_main_op_fn,
            assets_extra=None,
            as_text=False,
            clear_devices=True,
            strip_default_attrs=True,
            modes=None,
            **kwargs)

##################### Estimator ##########################

def export_all(
            export_dir_base,
            checkpoint_path,
            signature_defs_and_main_op_fn,
            assets_extra=None,
            as_text=False,
            clear_devices=True,
            strip_default_attrs=True,
            modes=None,
            **kwargs):
        r'''Build a SavedModel from variables in checkpoint.

        Args:
          export_dir_base: A string containing a directory to write the exported
              graph and checkpoints.
          checkpoint_path: A path to a checkpoint.
          signature_defs_and_main_op_fn: Function returns signature defs and main_op.
          assets_extra: A dict specifying how to populate the assets.extra directory
              within the exported SavedModel.  Each key should give the destination
              path (including the filename) relative to the assets.extra directory.
              The corresponding value gives the full path of the source file to be
              copied.  For example, the simple case of copying a single file without
              renaming it is specified as
              `{'my_asset_file.txt': '/path/to/my_asset_file.txt'}`.
          as_text: Whether or not to write the SavedModel proto in text format.
          clear_devices: Whether or not to clear the device field.
          strip_default_attrs: Whether or not to remove default-valued attributes
              from the NodeDefs.
          modes: List contains PREDICT, TRAIN or TEST.

        Returns:
          Export directory if it's chief.
        '''
        if HvdContext.get().rank != 0:
            return None

        export_dir = get_timestamped_export_dir(export_dir_base)
        with ops.Graph().as_default():
            with HvdContext.scope():
               # Build graph.
                signature_def_map = signature_defs_and_main_op_fn()
                main_op = None
                if isinstance(signature_def_map, (tuple, list)):
                    if len(signature_def_map) > 1:
                        main_op = signature_def_map[1]
                    signature_def_map = signature_def_map[0]
                if not main_op:
                    main_op = _monitored_session.Scaffold.default_local_init_op()
                if modes is None:
                    modes = [ModeKeys.PREDICT, ModeKeys.TRAIN, ModeKeys.EVAL]
                modes = [
                    m for m in modes
                    if SIGNATURE_KEY_MAP[m] in signature_def_map]
                signature_def_map = {
                    k: signature_def_map[k] for k in signature_def_map
                    if k in [SIGNATURE_KEY_MAP[m] for m in modes]}
                signature_tags = [EXPORT_TAG_MAP[m][0] for m in modes]