Searched defs:checkpoint_position (Results 1 – 2 of 2) sorted by relevance
103 def __init__(self, checkpoint_position): argument107 def checkpoint_position(self): member in CheckpointInitialValueCallable137 def __init__(self, checkpoint_position, shape=None, shard_info=None): argument156 def checkpoint_position(self): member in CheckpointInitialValue
482 def _queue_children_for_restoration(checkpoint_position, visit_queue): argument522 def _queue_slot_variables(checkpoint_position, visit_queue): argument