Searched refs:python_positions (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/checkpoint/ |
| H A D | restore.py | 162 python_positions = []
189 python_positions.append(self)
204 return (existing_restore_ops, named_saveables, python_positions,
335 (restore_ops, tensor_saveables, python_positions,
338 self._checkpoint.restore_saveables(tensor_saveables, python_positions))
439 python_positions = []
449 python_positions.extend(new_python_positions)
459 python_positions,
471 restore_ops, tensor_saveables, python_positions, registered_savers = (
477 python_positions = ()
[all …]
|
| H A D | checkpoint.py | 327 python_positions, argument
343 for position in python_positions:
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/python/ |
| H A D | d_checkpoint.py | 176 python_positions: List[restore_lib.CheckpointPosition],
196 if python_positions:
200 for position in python_positions:
|