Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tfx.components.trainer.rewriting.converters.rewrite_saved_model

View source on GitHub

Rewrites the provided SavedModel.

tfx.components.trainer.rewriting.converters.rewrite_saved_model(
    src, dst, rewriter_inst,
    dst_model_type=tfx.components.trainer.rewriting.rewriter.ModelType.SAVED_MODEL
)

Args:

  • src: location of the saved_model to rewrite.
  • dst: location of the rewritten saved_model.
  • rewriter_inst: the rewriter instance to invoke. Must inherit from rewriter.BaseRewriter.
  • dst_model_type: the rewriter.ModelType of the destination model.