Searched refs:slot_variable_position (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/training/ |
| D | optimizer.py | 1181 self, slot_variable_position, slot_name, variable): argument
1205 slot_variable_position.is_simple_variable()
1216 checkpoint_position=slot_variable_position)
1236 slot_variable_position.restore(slot_variable)
1244 slot_variable_position)
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | optimizer_v2.py | 1339 self, slot_variable_position, slot_name, variable): argument
1363 slot_variable_position.is_simple_variable()
1380 checkpoint_position=slot_variable_position)
1396 slot_variable_position.restore(slot_variable)
1404 slot_variable_position)
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer.py | 871 def _create_or_restore_slot_variable(self, slot_variable_position, slot_name, argument
874 slot_variable_position, slot_name, variable)
1147 def _create_or_restore_slot_variable(self, slot_variable_position, slot_name, argument
1150 slot_variable_position, slot_name, variable)
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | base.py | 278 slot_variable_position=CheckpointPosition(
303 slot_variable_position=CheckpointPosition(
|