|View source on GitHub|
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.