Searched refs:tensor_saveables (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/checkpoint/ |
| H A D | restore.py | 335 (restore_ops, tensor_saveables, python_positions,
338 self._checkpoint.restore_saveables(tensor_saveables, python_positions))
438 tensor_saveables = {}
448 tensor_saveables.update(new_tensor_saveables)
458 current_position.checkpoint.restore_saveables(tensor_saveables,
471 restore_ops, tensor_saveables, python_positions, registered_savers = (
476 tensor_saveables = {}
479 return restore_ops, tensor_saveables, python_positions, registered_savers
|
| H A D | checkpoint.py | 326 tensor_saveables, argument
348 if tensor_saveables or registered_savers:
350 tensor_saveables)
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/python/ |
| H A D | d_checkpoint.py | 175 tensor_saveables: Dict[str, saveable_object.SaveableObject],
205 if tensor_saveables:
207 tensor_saveables)
209 if set(tensor_saveables.keys()) != validated_names:
212 "expecting %s") % (tensor_saveables.keys(), validated_names))
|