Home
last modified time | relevance | path

Searched defs:num_accelerators (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/cluster_resolver/
Dcluster_resolver.py130 def num_accelerators(self, member in ClusterResolver
321 environment='', num_accelerators=None, argument
388 def num_accelerators(self, member in SimpleClusterResolver
611 def num_accelerators(self, member in UnionClusterResolver
Dslurm_cluster_resolver.py391 def num_accelerators(self, member in SlurmClusterResolver
Dtfconfig_cluster_resolver.py140 def num_accelerators(self, member in TFConfigClusterResolver
Dsagemaker_cluster_resolver.py150 def num_accelerators(self, task_type=None, task_id=None, config_proto=None): member in SageMakerClusterResolver
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/tpu/
Dtpu_cluster_resolver.py343 def num_accelerators(self, member in TPUClusterResolver