Home
last modified time | relevance | path

Searched refs:run_restore_ops (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/python/checkpoint/
Dcheckpoint_test.py328 status.assert_consumed().run_restore_ops()
478 status.assert_existing_objects_matched().run_restore_ops()
508 status.run_restore_ops()
532 first_status.run_restore_ops()
536 second_status.run_restore_ops()
550 first_status.run_restore_ops()
554 second_status.run_restore_ops()
607 status.run_restore_ops()
653 status.run_restore_ops()
663 status.run_restore_ops()
[all …]
Dcheckpoint_with_v1_optimizers_test.py70 no_slot_status.run_restore_ops()
91 slot_status.run_restore_ops()
175 status.assert_consumed().run_restore_ops()
248 status.assert_consumed().run_restore_ops()
Dcheckpoint.py695 def run_restore_ops(self, session=None): member in _LoadStatus
729 status.run_restore_ops(session=session)
880 def run_restore_ops(self, session=None): member in CheckpointLoadStatus
917 self.run_restore_ops(session=session)
957 def run_restore_ops(self, session=None): member in InitializationOnlyStatus
1088 def run_restore_ops(self, session=None): member in NameBasedSaverStatus
1101 self.run_restore_ops(session=session)
Dcheckpoint_management_test.py479 checkpoint.restore(manager.latest_checkpoint).run_restore_ops()
603 init_fn=lambda: checkpoint.restore(init_path).run_restore_ops())
/external/tensorflow/tensorflow/python/trackable/
Dpython_state_test.py169 loader.restore(first_save_path).run_restore_ops()
171 loader.restore(second_save_path).run_restore_ops()
Dbase_delegate_test.py65 ckpt2.restore(prefix_tensor).assert_consumed().run_restore_ops()
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dcheckpoint_test.py308 checkpoint.restore(save_path).run_restore_ops()
338 checkpoint.restore(save_path).run_restore_ops()
355 checkpoint.restore(save_path).run_restore_ops()
358 checkpoint.restore(save_path).run_restore_ops()
581 checkpoint.restore(save_path).run_restore_ops()
/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy_model_parallelism_test.py166 status.run_restore_ops(sess) # must run restore op in non-eager mode.
357 status.run_restore_ops(sess) # must run restore op in non-eager mode.
Ddistributed_variable_test.py135 checkpoint.restore(save_path).assert_consumed().run_restore_ops()
/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale_optimizer_test.py302 status.run_restore_ops()
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dlayer_test.py266 checkpoint.restore(save_path).assert_consumed().run_restore_ops()
/external/tensorflow/tensorflow/python/kernel_tests/data_structures/
Dlookup_ops_test.py2019 restore_checkpoint.restore(save_path).run_restore_ops()
3571 checkpoint.restore(save_path).run_restore_ops()
/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
Drnn_cell_test.py2902 checkpoint.restore(save_path).assert_consumed().run_restore_ops()