def __init__(self, cluster_server, trainer_master, bridge, role, model_fn, is_chief=False):
    super(SparseFLEstimator, self).__init__(cluster_server, trainer_master, bridge, role, model_fn, is_chief)
    self._bias_slot_configs = None
    self._vec_slot_configs = None
    self._slot_configs = None
    try:
        ps_indices = cluster_server.cluster_spec.task_indices('ps')
    except ValueError:
        ps_indices = None
    finally:
        self._embedding_devices = ([None] if (not ps_indices) else [('/job:ps/task:%d' % i) for i in ps_indices])
    self._num_shards = len(self._embedding_devices)
def __init__(self, cluster_server, trainer_master, bridge, role, model_fn, is_chief=False):
    super(SparseFLEstimator, self).__init__(cluster_server, trainer_master, bridge, role, model_fn, is_chief)
    self._bias_slot_configs = None
    self._vec_slot_configs = None
    self._slot_configs = None
    try:
        ps_indices = cluster_server.cluster_spec.task_indices('ps')
    except ValueError:
        ps_indices = None
    finally:
        self._embedding_devices = ([None] if (not ps_indices) else [('/job:ps/task:%d' % i) for i in ps_indices])
    self._num_shards = len(self._embedding_devices)