Method __init__

tensorflow/python/distribute/hvd_strategy.py:547–553

(self, *args, **kwargs)

        '''

        def __init__(self, *args, **kwargs):
            self._rank = HvdContext.get().rank
            self._world_size = HvdContext.get().world_size
            kwargs['sharded'] = True
            kwargs['allow_empty'] = True
            with ops.device('/cpu:0'):
                super().__init__(*args, **kwargs)

        @property
        def rank(self):