Home
last modified time | relevance | path

Searched refs:as_dict (Results 1 – 25 of 33) sorted by relevance

12

/external/tensorflow/tensorflow/python/distribute/
Dmulti_worker_util.py140 cluster_spec = normalize_cluster_spec(cluster_spec).as_dict()
168 if not cluster_spec.as_dict():
189 cluster_spec = normalize_cluster_spec(cluster_spec).as_dict()
227 cluster_spec = normalize_cluster_spec(cluster_spec).as_dict()
Dparameter_server_strategy_v2.py454 self._num_workers = len(cluster_spec.as_dict().get("worker", ()))
455 self._num_ps = len(cluster_spec.as_dict().get("ps", ()))
518 self._num_ps = len(cluster_resolver.cluster_spec().as_dict().get("ps", []))
Ddistribute_coordinator.py99 return len(cluster_spec.as_dict().get(_TaskType.WORKER, [])) + len(
100 cluster_spec.as_dict().get(_TaskType.CHIEF, []))
374 cluster_spec).as_dict()
473 for task_id in range(len(cluster_spec.as_dict().get(task_type, []))):
783 cluster_spec.as_dict(), task_type, task_id, environment, rpc_layer)
Dmulti_worker_util_test.py31 server_lib.ClusterSpec(lhs).as_dict(),
32 server_lib.ClusterSpec(rhs).as_dict())
Destimator_training.py46 return len(cluster_spec.as_dict().get(PS, []))
55 return (len(cluster_spec.as_dict().get(WORKER, [])) + len(
56 cluster_spec.as_dict().get(chief_task_type, [])))
Dmulti_worker_test_base_test.py74 cluster_spec = self._cluster.cluster_resolver.cluster_spec().as_dict()
Dparameter_server_strategy.py222 assert cluster_spec.as_dict()
248 num_ps_replicas = len(cluster_spec.as_dict().get("ps", []))
277 "variable_device = %r", cluster_spec.as_dict(), task_type, task_id,
Dmirrored_strategy.py91 for task_id in range(len(cluster_spec.as_dict().get(task_type, []))):
185 if tfconfig.cluster_spec().as_dict():
316 if TFConfigClusterResolver().cluster_spec().as_dict():
Dparameter_server_strategy_test.py376 num_workers = len(d.extended._cluster_spec.as_dict().get(WORKER))
377 if 'chief' in d.extended._cluster_spec.as_dict():
442 num_workers = len(d.extended._cluster_spec.as_dict().get(WORKER))
443 if CHIEF in d.extended._cluster_spec.as_dict():
Dcollective_all_reduce_strategy.py335 if cluster_resolver.cluster_spec().as_dict():
516 "communication = %s", cluster_spec.as_dict(), task_type, task_id,
/external/tensorflow/tensorflow/python/training/
Ddevice_setter.py204 cluster_spec = cluster.as_dict()
206 cluster_spec = server_lib.ClusterSpec(cluster).as_dict()
Dserver_lib.py335 key_values = self.as_dict()
341 def as_dict(self): member in ClusterSpec
Dserver_lib_test.py517 server_lib.ClusterSpec(cluster_spec.as_dict()).as_cluster_def())
539 server_lib.ClusterSpec(cluster_spec.as_dict()).as_cluster_def())
552 self.assertEqual(original_dict, cluster_spec.as_dict())
Ddevice_setter_test.py122 device_setter.replica_device_setter(cluster=self._cluster_spec.as_dict(
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.train.-cluster-spec.pbtxt18 name: "as_dict"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.train.-cluster-spec.pbtxt18 name: "as_dict"
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/tpu/
Dtpu_cluster_resolver_test.py123 server_lib.ClusterSpec(cluster_spec.as_dict()).as_cluster_def())
457 }).as_dict(),
458 cluster_resolver.cluster_spec().as_dict())
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/
Dcluster_resolver.py529 cluster_dict = cluster_spec.as_dict()
546 cluster_dict = cluster_spec.as_dict()
Dsagemaker_cluster_resolver_test.py44 server_lib.ClusterSpec(cluster_spec.as_dict()).as_cluster_def())
Dkubernetes_cluster_resolver_test.py73 cluster_spec.as_dict()).as_cluster_def())
Dslurm_cluster_resolver_test.py62 server_lib.ClusterSpec(cluster_spec.as_dict()).as_cluster_def())
Dtfconfig_cluster_resolver_test.py46 server_lib.ClusterSpec(cluster_spec.as_dict()).as_cluster_def())
Dgce_cluster_resolver_test.py41 server_lib.ClusterSpec(cluster_spec.as_dict()).as_cluster_def())
/external/chromium-trace/catapult/common/py_utils/py_utils/
Dmemory_debug.py34 pinfo = p.as_dict(attrs=['pid', 'name', 'memory_info'])
/external/chromium-trace/catapult/devil/devil/android/tools/
Ddevice_recovery.py44 pinfo = p.as_dict(attrs=['pid', 'name', 'cmdline'])

12