HomeSort by relevance Sort by last modified time
    Searched refs:per_device (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/python/training/
saver.py 212 def _AddShardedSaveOpsForV2(self, checkpoint_prefix, per_device):
222 per_device: A list of (device, BaseSaverBuilder.VarToSave) pairs, as
257 num_shards = len(per_device)
262 for shard, (device, saveables) in enumerate(per_device):
282 def _AddShardedSaveOps(self, filename_tensor, per_device):
287 per_device: A list of (device, BaseSaverBuilder.SaveableObject) pairs, as
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,
    [all...]
  /external/tensorflow/tensorflow/python/grappler/
cluster.i 412 PyObject* per_device = PyList_New(usage.live_tensors.size());
421 PyList_SetItem(per_device, i, live);
426 PyTuple_SetItem(ret, 1, per_device);

Completed in 463 milliseconds