Defined in tensorflow/contrib/cluster_resolver/python/training/

Library Imports for Cluster Resolvers.


tpu_cluster_resolver module: Stub file for TPUClusterResolver to maintain backwards compatibility.


class ClusterResolver: Abstract class for all implementations of ClusterResolvers.

class GceClusterResolver: Cluster Resolver for Google Compute Engine.

class KubernetesClusterResolver: Cluster Resolver for Kubernetes.

class SimpleClusterResolver: Simple implementation of ClusterResolver that accepts a ClusterSpec.

class SlurmClusterResolver: Cluster Resolver for system with Slurm workload manager.

class TFConfigClusterResolver: Implementation of a ClusterResolver which reads the TF_CONFIG EnvVar.

class TPUClusterResolver: Cluster Resolver for Google Cloud TPUs.

class UnionClusterResolver: Performs a union on underlying ClusterResolvers.