TensorFlow 2 version |
Library imports for ClusterResolvers.
This library contains all implementations of ClusterResolvers.
ClusterResolvers are a way of specifying cluster information for distributed
execution. Built on top of existing ClusterSpec
framework, ClusterResolvers
are a way for TensorFlow to communicate with various cluster management
systems (e.g. GCE, AWS, etc...).
Classes
class ClusterResolver
: Abstract class for all implementations of ClusterResolvers.
class GCEClusterResolver
: ClusterResolver for Google Compute Engine.
class KubernetesClusterResolver
: ClusterResolver for Kubernetes.
class SimpleClusterResolver
: Simple implementation of ClusterResolver that accepts a ClusterSpec.
class SlurmClusterResolver
: ClusterResolver 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 UnionResolver
: Performs a union on underlying ClusterResolvers.