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

  /external/tensorflow/tensorflow/python/training/
saver.py 329 def _AddShardedSaveOpsForV2(self, checkpoint_prefix, per_device):
339 per_device: A list of (device, BaseSaverBuilder.VarToSave) pairs, as
374 num_shards = len(per_device)
379 for shard, (device, saveables) in enumerate(per_device):
399 def _AddShardedSaveOps(self, filename_tensor, per_device):
404 per_device: A list of (device, BaseSaverBuilder.SaveableObject) pairs, as
411 return self._AddShardedSaveOpsForV2(filename_tensor, per_device)
413 num_shards = len(per_device)
416 for shard, (device, saveables) in enumerate(per_device):
476 def _AddShardedRestoreOps(self, filename_tensor, per_device,
    [all...]
  /system/extras/boottime_tools/io_analysis/
check_io_trace_all.py 167 per_device = self.total_reads_per_device.get(devNum)
168 if not per_device:
174 per_device = self.total_writes_per_device.get(devNum)
175 if not per_device:
  /external/tensorflow/tensorflow/python/grappler/
cluster.i 404 PyObject* per_device = PyList_New(usage.live_tensors.size());
413 PyList_SetItem(per_device, i, live);
418 PyTuple_SetItem(ret, 1, per_device);

Completed in 182 milliseconds