TensorFlow 2.0 Beta is available Learn more

tf.distribute.cluster_resolver.UnionResolver

TensorFlow 2.0 version View source on GitHub

Class UnionResolver

Performs a union on underlying ClusterResolvers.

Inherits From: ClusterResolver

Aliases:

  • Class tf.compat.v1.distribute.cluster_resolver.UnionResolver
  • Class tf.compat.v2.distribute.cluster_resolver.UnionResolver
  • Class tf.contrib.cluster_resolver.UnionClusterResolver
  • Class tf.contrib.cluster_resolver.python.training.UnionClusterResolver

This class performs a union given two or more existing ClusterResolvers. It merges the underlying ClusterResolvers, and returns one unified ClusterSpec when cluster_spec is called. The details of the merge function is documented in the cluster_spec function.

For additional Cluster Resolver properties such as task type, task index, rpc layer, environment, etc..., we will return the value from the first ClusterResolver in the union.

__init__

View source

__init__(
    *args,
    **kwargs
)

Initializes a UnionClusterResolver with other ClusterResolvers.

Args:

  • *args: ClusterResolver objects to be unionized.
  • **kwargs: rpc_layer - (Optional) Override value for the RPC layer used by TensorFlow. task_type - (Optional) Override value for the current task type. task_id - (Optional) Override value for the current task index.

Raises:

  • TypeError: If any argument is not a subclass of ClusterResolvers.
  • ValueError: If there are no arguments passed.

Properties

environment

rpc_layer

task_id

task_type

Methods

cluster_spec

View source

cluster_spec()

Returns a union of all the ClusterSpecs from the ClusterResolvers.

Returns:

A ClusterSpec containing host information merged from all the underlying ClusterResolvers.

Raises:

  • KeyError: If there are conflicting keys detected when merging two or more dictionaries, this exception is raised.

If all underlying ClusterSpecs expose the set of workers as lists, we will concatenate the lists of workers, starting with the list of workers from the first ClusterResolver passed into the constructor.

If any of the ClusterSpecs expose the set of workers as a dict, we will treat all the sets of workers as dicts (even if they are returned as lists) and will only merge them into a dict if there is no conflicting keys. If there is a conflicting key, we will raise a KeyError.

master

View source

master(
    task_type=None,
    task_id=None,
    rpc_layer=None
)

Returns the master address to use when creating a session.

This usually returns the master from the first ClusterResolver passed in, but you can override this by specifying the task_type and task_id.

Args:

  • task_type: (Optional) The type of the TensorFlow task of the master.
  • task_id: (Optional) The index of the TensorFlow task of the master.
  • rpc_layer: (Optional) The RPC protocol for the given cluster.

Returns:

The name or URL of the session master.

num_accelerators

View source

num_accelerators(
    task_type=None,
    task_id=None,
    config_proto=None
)