Home
last modified time | relevance | path

Searched refs:checkpoint_position (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/python/trackable/
Dbase.py103 def __init__(self, checkpoint_position): argument
104 self._checkpoint_position = checkpoint_position
107 def checkpoint_position(self): member in CheckpointInitialValueCallable
137 def __init__(self, checkpoint_position, shape=None, shard_info=None): argument
145 self.wrapped_value = checkpoint_position.value_tensors(
147 self._checkpoint_position = checkpoint_position
156 def checkpoint_position(self): member in CheckpointInitialValue
530 for checkpoint_position in deferred_dependencies_list:
531 if not checkpoint_position.is_simple_variable():
537 checkpoint_position = max(
[all …]
/external/tensorflow/tensorflow/python/checkpoint/
Drestore.py482 def _queue_children_for_restoration(checkpoint_position, visit_queue): argument
485 trackable = checkpoint_position.trackable
486 for child in checkpoint_position.object_proto.children:
487 child_position = checkpoint_position.create_child_position(child.node_id)
522 def _queue_slot_variables(checkpoint_position, visit_queue): argument
524 trackable = checkpoint_position.trackable
525 checkpoint = checkpoint_position.checkpoint
527 checkpoint_position.proto_id, ())):
529 checkpoint_position.create_slot_variable_position(
536 checkpoint_position.proto_id, ()):
[all …]
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.__internal__.tracking.-checkpoint-initial-value-callable.pbtxt6 name: "checkpoint_position"
15 argspec: "args=[\'self\', \'checkpoint_position\'], varargs=None, keywords=None, defaults=None"
Dtensorflow.__internal__.tracking.-checkpoint-initial-value.pbtxt13 name: "checkpoint_position"
46 …argspec: "args=[\'self\', \'checkpoint_position\', \'shape\', \'shard_info\'], varargs=None, keywo…
/external/tensorflow/tensorflow/python/training/
Doptimizer.py1271 for checkpoint_position in deferred_restorations:
1272 checkpoint_position.restore(slot_variable)
1310 checkpoint_position=slot_variable_position)
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Doptimizer_v2.py1339 for checkpoint_position in deferred_restorations:
1340 checkpoint_position.restore(slot_variable)
1384 checkpoint_position=slot_variable_position)
/external/tensorflow/tensorflow/python/distribute/
Ddistribute_lib.py2147 "initial_value"].checkpoint_position.restore_uid
2152 "initial_value"].checkpoint_position.restore_uid
2162 "initial_value"].func.checkpoint_position.restore_uid
/external/tensorflow/tensorflow/python/ops/
Dvariables.py1758 self._update_uid = initial_value.checkpoint_position.restore_uid
1794 self._update_uid = initial_value.checkpoint_position.restore_uid
Dresource_variable_ops.py1820 self._update_uid = initial_value.checkpoint_position.restore_uid
/external/tensorflow/tensorflow/python/eager/
Ddef_function.py298 self._update_uid = initial_value.checkpoint_position.restore_uid