Home | History | Annotate | Download | only in python

Lines Matching refs:Saver

1 """Tests for eager mode Saver."""
22 from tensorflow.contrib.eager.python import saver as _saver
53 saver = _saver.Saver([v1])
54 saver.save(ckpt_prefix)
58 saver.restore(ckpt_prefix)
71 saver = _saver.Saver([v1_first_graph, v1_second_graph])
74 saver.save(ckpt_prefix)
81 saver = _saver.Saver([v1, v1])
83 saver.save(ckpt_prefix)
96 saver = _saver.Saver({'ckpt/v1': v1, 'ckpt/v2': v2})
97 saver.save(ckpt_prefix)
103 saver.restore(ckpt_prefix)
108 saver = _saver.Saver([v1, v2]).restore(ckpt_prefix)
127 saver = _saver.Saver([v1])
128 saver.save(ckpt_prefix)
131 saver = _saver.Saver([v1])
145 saver = _saver.Saver([v])
146 saver.save(ckpt_prefix)
168 _saver.Saver(model.variables).save(ckpt_prefix)
176 _saver.Saver(model.variables).restore(ckpt_prefix)
182 _saver.Saver(model.variables).save(ckpt_prefix)
209 saver = _saver.Saver(optimizer_variables + [v])
210 checkpoint_path = saver.save(self.get_temp_dir())
215 saver.restore(checkpoint_path)