Searched defs:per_device (Results 1 – 2 of 2) sorted by relevance
212 def _AddShardedSaveOpsForV2(self, checkpoint_prefix, per_device): argument282 def _AddShardedSaveOps(self, filename_tensor, per_device): argument358 def _AddShardedRestoreOps(self, filename_tensor, per_device, argument
412 PyObject* per_device = PyList_New(usage.live_tensors.size()); in TF_DeterminePeakMemoryUsage() local