Searched defs:checkpoint_position (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | base.py | 75 def __init__(self, checkpoint_position): argument
79 def checkpoint_position(self): member in CheckpointInitialValueCallable
109 def __init__(self, checkpoint_position, shape=None, shard_info=None): argument
128 def checkpoint_position(self): member in CheckpointInitialValue
959 def _restore_from_checkpoint_position(self, checkpoint_position): argument
984 def _single_restoration_from_checkpoint_position(self, checkpoint_position, argument
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer.py | 158 def _restore_from_checkpoint_position(self, checkpoint_position): argument
162 def _single_restoration_from_checkpoint_position(self, checkpoint_position, argument
|