r'''Saves sharded variables.
(self, *args, **kwargs)
| 571 | super()._build(*args, **kwargs) |
| 572 | |
| 573 | def save(self, *args, **kwargs): |
| 574 | r'''Saves sharded variables. |
| 575 | ''' |
| 576 | if self._world_size <= 1: |
| 577 | super().save(*args, **kwargs) |
| 578 | return |
| 579 | |
| 580 | write_meta_graph = ( |
| 581 | kwargs.pop('write_meta_graph', True) |
| 582 | and self._rank == 0) |
| 583 | kwargs['write_meta_graph'] = write_meta_graph |
| 584 | write_state = kwargs.pop('write_state', True) and self._rank == 0 |
| 585 | kwargs['write_state'] = write_state |
| 586 | super().save(*args, **kwargs) |
| 587 | |
| 588 | def export_meta_graph(self, filename=None, **kwargs): |
| 589 | if self._rank == 0: |