Searched refs:slot_variable_position (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/checkpoint/ |
| D | restore.py | 407 slot_variable_position = CheckpointPosition(
411 slot_variable_position=slot_variable_position,
416 slot_variable_position.bind_object(slot_variable)):
417 return slot_variable_position, slot_variable
528 slot_variable_position, slot_variable = (
533 if slot_variable_position is not None:
534 visit_queue.append((slot_variable_position, slot_variable))
554 slot_variable_position, slot_variable = (
558 if slot_variable_position is not None:
559 visit_queue.append((slot_variable_position, slot_variable))
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | optimizer_v2.py | 1343 self, slot_variable_position, slot_name, variable): argument
1367 slot_variable_position.is_simple_variable()
1384 checkpoint_position=slot_variable_position)
1389 shape=slot_variable_position.value_shape())
1400 slot_variable_position.restore(slot_variable)
1408 slot_variable_position)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | optimizer.py | 1275 self, slot_variable_position, slot_name, variable): argument
1299 slot_variable_position.is_simple_variable()
1310 checkpoint_position=slot_variable_position)
1330 slot_variable_position.restore(slot_variable)
1338 slot_variable_position)
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer.py | 789 def _create_or_restore_slot_variable(self, slot_variable_position, slot_name, argument
792 slot_variable_position, slot_name, variable)
1086 def _create_or_restore_slot_variable(self, slot_variable_position, slot_name, argument
1089 slot_variable_position, slot_name, variable)
|