Searched refs:assert_consumed (Results 1 – 11 of 11) sorted by relevance
371 checkpoint.restore(save_path).assert_consumed().initialize_or_restore()377 checkpoint.restore(save_path).assert_consumed().initialize_or_restore()424 status = root_trackable.restore(save_path=save_path).assert_consumed()440 status.assert_consumed()451 status.assert_consumed()459 status.assert_consumed()517 status.assert_consumed()521 status.assert_consumed()603 status.assert_consumed().run_restore_ops()653 status.assert_consumed()[all …]
216 status = root_trackable.restore(save_path=save_path).assert_consumed()236 status.assert_consumed()246 status.assert_consumed()255 status.assert_consumed()340 status.assert_consumed()373 status.assert_consumed()377 status.assert_consumed()452 status.assert_consumed()533 no_slot_status.assert_consumed()536 no_slot_status.assert_consumed()[all …]
69 restored.restore(save_path).assert_consumed()84 status.assert_consumed()
464 def assert_consumed(self): member in _LoadStatus537 def assert_consumed(self): member in CheckpointLoadStatus683 def assert_consumed(self): member in InitializationOnlyStatus758 def assert_consumed(self): member in NameBasedSaverStatus779 return self.assert_consumed()787 return self.assert_consumed()
184 checkpoint.restore(save_path).assert_consumed().initialize_or_restore()
234 status = root_trackable.restore(save_path=save_path).assert_consumed()266 status.assert_consumed()320 status.assert_consumed()322 status.assert_consumed()399 status.assert_consumed()470 no_slot_status.assert_consumed()473 no_slot_status.assert_consumed()478 slot_status.assert_consumed()498 slot_status.assert_consumed()578 status.assert_consumed().run_restore_ops()[all …]
90 save_path).assert_consumed().run_restore_ops()99 status.assert_consumed()106 status.assert_consumed().run_restore_ops()
63 status.assert_consumed().run_restore_ops()
325 checkpoint.restore(save_path).assert_consumed().run_restore_ops()333 status.assert_consumed().run_restore_ops()
147 checkpoint.restore(save_path).assert_consumed().run_restore_ops()156 status.assert_consumed().run_restore_ops()283 checkpoint.restore(save_path).assert_consumed().run_restore_ops()292 status.assert_consumed().run_restore_ops()
2812 checkpoint.restore(save_path).assert_consumed().run_restore_ops()