Saved Model Builder¶

AutoDist SavedModelBuilder.

class SavedModelBuilder(export_dir)[source]¶

Bases: tensorflow.python.saved_model.builder_impl.SavedModelBuilder

Wrapper of SavedModelBuilder.

add_meta_graph_and_variables(sess, tags, signature_def_map=None, assets_collection=None, legacy_init_op=None, clear_devices=False, main_op=None, strip_default_attrs=False, saver=None, train_op=None)[source]¶: Save graph variables and metagraph.