Module: tf.distribute.cluster_resolver

Public API for tf._api.v2.distribute.cluster_resolver namespace


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 all attributes.

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.