Searched refs:per_device (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/training/ |
| D | saver.py | 209 def _AddShardedSaveOpsForV2(self, checkpoint_prefix, per_device): argument
254 num_shards = len(per_device)
259 for shard, (device, saveables) in enumerate(per_device):
279 def _AddShardedSaveOps(self, filename_tensor, per_device): argument
291 return self._AddShardedSaveOpsForV2(filename_tensor, per_device)
293 num_shards = len(per_device)
296 for shard, (device, saveables) in enumerate(per_device):
355 def _AddShardedRestoreOps(self, filename_tensor, per_device, argument
372 for shard, (device, saveables) in enumerate(per_device):
401 per_device = collections.defaultdict(lambda: [])
[all …]
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cluster_wrapper.cc | 320 std::vector<MemoryUsage> per_device; in PYBIND11_MODULE() local
323 per_device.push_back(std::make_tuple( in PYBIND11_MODULE()
328 result[device.first] = std::make_tuple(usage.used_memory, per_device); in PYBIND11_MODULE()
|