Searched refs:_AddShardedSaveOps (Results 1 – 1 of 1) sorted by relevance
282 def _AddShardedSaveOps(self, filename_tensor, per_device): member in BaseSaverBuilder503 save_tensor = self._AddShardedSaveOps(filename_tensor, per_device)