Searched refs:before_save (Results 1 – 12 of 12) sorted by relevance
53 def before_save(self): member in SaveableHook
166 self._before_save_callbacks.append(saveable.before_save)
202 def before_save(self): member in SaverTest.test_callbacks_run.DummyHook
13 name: "before_save"
145 mock_listener.before_save.assert_called()
167 l.before_save(session, step)
1467 def before_save(self): member in TPUEmbeddingSaveable
502 def before_save(self, session, global_step_value): member in CheckpointSaverListener616 l.before_save(session, step)
72 def before_save(self, session, global_step): member in MockCheckpointSaverListener
442 before_save = self.evaluate(v.read_value())453 self.assertNotAllClose(before_save, after_assign)459 self.assertAllClose(before_save, after_restore)