Searched refs:per_device (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/training/ |
| D | saver.py | 212 def _AddShardedSaveOpsForV2(self, checkpoint_prefix, per_device): argument
257 num_shards = len(per_device)
262 for shard, (device, saveables) in enumerate(per_device):
282 def _AddShardedSaveOps(self, filename_tensor, per_device): argument
294 return self._AddShardedSaveOpsForV2(filename_tensor, per_device)
296 num_shards = len(per_device)
299 for shard, (device, saveables) in enumerate(per_device):
358 def _AddShardedRestoreOps(self, filename_tensor, per_device, argument
375 for shard, (device, saveables) in enumerate(per_device):
404 per_device = collections.defaultdict(lambda: [])
[all …]
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cluster.i | 412 PyObject* per_device = PyList_New(usage.live_tensors.size()); in TF_DeterminePeakMemoryUsage() local
421 PyList_SetItem(per_device, i, live); in TF_DeterminePeakMemoryUsage()
426 PyTuple_SetItem(ret, 1, per_device); in TF_DeterminePeakMemoryUsage()
|