voctoweb | voctoweb – the frontend and backend software behind media

def _initialize_local(self, cluster_resolver, devices=None):
    """Initializes the object for local training."""
    self._is_chief = True
    self._num_workers = 1

    if ops.executing_eagerly_outside_functions():
      try:
        context.contex  

def get_workers_list(cluster_resolver):
  """Returns a comma separated list of TPU worker host:port pairs.

  Gets cluster_spec from cluster_resolver. Use the worker's task indices to
  obtain and return a list of host:port pairs.

  Args:
    cluste  

def _multi_worker_concat(v, strategy):
  """Order PerReplica objects for CollectiveAllReduceStrategy and concat."""
  replicas = strategy.gather(v, axis=0)
  # v might not have the same shape on different replicas
  if _is_per_replica_instance(v):