Searched defs:per_device (Results 1 – 2 of 2) sorted by relevance
209 def _AddShardedSaveOpsForV2(self, checkpoint_prefix, per_device): argument279 def _AddShardedSaveOps(self, filename_tensor, per_device): argument355 def _AddShardedRestoreOps(self, filename_tensor, per_device, argument
320 std::vector<MemoryUsage> per_device; in PYBIND11_MODULE() local