Searched refs:per_device (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/training/ |
| D | saver.py | 208 def _AddShardedSaveOpsForV2(self, checkpoint_prefix, per_device): argument
262 num_shards = len(per_device)
267 for shard, (device, saveables) in enumerate(per_device):
287 def _AddShardedSaveOps(self, filename_tensor, per_device): argument
299 return self._AddShardedSaveOpsForV2(filename_tensor, per_device)
301 num_shards = len(per_device)
304 for shard, (device, saveables) in enumerate(per_device):
363 def _AddShardedRestoreOps(self, filename_tensor, per_device, argument
380 for shard, (device, saveables) in enumerate(per_device):
409 per_device = collections.defaultdict(lambda: [])
[all …]
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cluster_wrapper.cc | 321 std::vector<MemoryUsage> per_device; in PYBIND11_MODULE() local
324 per_device.push_back(std::make_tuple( in PYBIND11_MODULE()
329 result[device.first] = std::make_tuple(usage.used_memory, per_device); in PYBIND11_MODULE()
|