Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/cluster_resolver/
H A 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
H A Dslurm_cluster_resolver.py391 def num_accelerators(self, member in SlurmClusterResolver
H A Dtfconfig_cluster_resolver.py140 def num_accelerators(self, member in TFConfigClusterResolver
H A Dsagemaker_cluster_resolver.py150 def num_accelerators(self, task_type=None, task_id=None, config_proto=None): member in SageMakerClusterResolver
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/cluster_resolver/tpu/
H A Dtpu_cluster_resolver.py343 def num_accelerators(self, member in TPUClusterResolver