Home
last modified time | relevance | path

Searched refs:task_type (Results 1 – 25 of 85) sorted by relevance

1234

/external/tensorflow/tensorflow/python/distribute/
Dmulti_worker_util.py49 def task_count(cluster_spec, task_type): argument
51 return cluster_spec.num_tasks(task_type)
57 task_type, argument
90 if task_type not in allowed_task_types:
93 task_type, allowed_task_types))
95 if (task_type and task_type not in cluster_spec.jobs and
96 task_type != "evaluator"):
97 raise ValueError("`task_type` %r not found in cluster_spec." % task_type)
106 if task_type in cluster_spec.jobs and task_id >= task_count(
107 cluster_spec, task_type):
[all …]
Ddistribute_coordinator.py115 task_type, argument
139 self._task_type = task_type
151 self._cluster_spec, self.task_type, self.task_id)
179 task_type = _TaskType.CHIEF
183 task_type = _TaskType.WORKER
186 task_type = self._task_type
192 return prefix + self._cluster_spec.job_tasks(task_type)[task_id or 0]
286 def task_type(self): member in _WorkerContext
329 task_type, argument
339 if task_type == _TaskType.EVALUATOR:
[all …]
Dmulti_process_runner.py251 def _continuously_readline_from_sub(self, pipe_r, task_type, task_id): argument
255 task_string = '[{}-{}]:'.format(task_type, task_id)
265 task_type, argument
278 task_type=task_type,
308 self._processes[(task_type, task_id)] = p
309 self._terminated.discard((task_type, task_id))
315 args=(pipe_r, task_type, task_id))
336 for task_type, addresses in self._cluster_spec.items():
338 self._start_subprocess_and_reading_thread(task_type, task_id)
394 for task_type, addresses in self._cluster_spec.items():
[all …]
Dmulti_worker_test_base.py233 task_type = cluster_resolver.task_type
239 'task_id = %r, rpc_layer = %r', cluster_spec, task_type, task_id,
253 job_name=task_type,
259 start_event = start_events[task_type][task_id]
262 finish_event = finish_events[task_type][task_id]
277 for task_type, task_addresses in self._cluster_spec.items():
278 self._start_events[task_type] = []
279 self._finish_events[task_type] = []
281 self._start_events[task_type].append(self._mpr_manager.Event())
282 self._finish_events[task_type].append(self._mpr_manager.Event())
[all …]
Ddistribute_coordinator_test.py111 task_type=None, argument
130 if (cluster_spec and task_type and task_id is not None and
133 if task_type in ["chief", "worker"]:
135 ["/job:%s/task:%d" % (task_type, task_id), "/job:ps"])
260 for task_type in cluster_spec.keys():
261 threads[task_type] = []
262 for task_id in range(len(cluster_spec[task_type])):
267 task_type=task_type,
270 threads[task_type].append(t)
374 task_type = str(context.task_type)
[all …]
Dcollective_all_reduce_strategy_test.py70 task_type=None, argument
77 if cluster_spec and task_type and task_id is not None:
80 task_type=task_type,
83 target = 'grpc://' + cluster_spec[task_type][task_id]
105 def _get_test_object(self, task_type, task_id, num_gpus=0): argument
108 task_type=task_type,
113 def _test_minimize_loss_graph(self, task_type, task_id, num_gpus): argument
114 d, master_target, config = self._get_test_object(task_type, task_id,
185 def _test_variable_initialization(self, task_type, task_id, num_gpus): argument
187 task_type, task_id, num_gpus)
[all …]
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/
Dtfconfig_cluster_resolver.py85 task_type=None, argument
100 self._task_type = task_type
106 def task_type(self): member in TFConfigClusterResolver
121 @task_type.setter
122 def task_type(self, task_type): argument
123 self._task_type = task_type
145 task_type=None, argument
148 task_type = self.task_type if task_type is None else task_type
151 task_type, task_id, config_proto)
164 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
[all …]
Dsagemaker_cluster_resolver.py95 task_type=None, argument
109 self._task_type = task_type
116 def task_type(self): member in SageMakerClusterResolver
131 @task_type.setter
132 def task_type(self, task_type): argument
133 self._task_type = task_type
154 def num_accelerators(self, task_type=None, task_id=None, config_proto=None): argument
155 task_type = self.task_type if task_type is None else task_type
158 self).num_accelerators(task_type, task_id, config_proto)
171 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
[all …]
Dgce_cluster_resolver.py73 task_type='worker', argument
109 self._task_type = task_type
174 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
175 task_type = task_type if task_type is not None else self._task_type
178 if task_type is not None and task_id is not None:
179 master = self.cluster_spec().task_address(task_type, task_id)
188 def task_type(self): member in GCEClusterResolver
195 @task_type.setter
196 def task_type(self, task_type): member in GCEClusterResolver
Dslurm_cluster_resolver.py263 self.task_type = None
342 for task_type, num_tasks in sorted(self._jobs.items()):
345 self._cluster_allocation[task_type] = (
349 self.task_type = task_type
371 return self.task_type, self.task_id
373 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
385 task_type = task_type if task_type is not None else self.task_type
388 if task_type is not None and task_id is not None:
390 self.cluster_spec().task_address(task_type, task_id),
396 task_type=None, argument
[all …]
Dkubernetes_cluster_resolver.py117 self.task_type = None
121 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
139 task_type = task_type if task_type is not None else self.task_type
142 if task_type is not None and task_id is not None:
144 self.cluster_spec().task_address(task_type, task_id),
Dcluster_resolver_test.py39 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
115 self.assertEqual(resolver.num_accelerators(task_type="worker1", task_id=0),
117 self.assertEqual(resolver.num_accelerators(task_type="worker2", task_id=3),
119 self.assertEqual(resolver.num_accelerators(task_type="worker2", task_id=4),
162 simple_resolver = SimpleClusterResolver(base_cluster_spec, task_type="ps",
167 self.assertEqual(simple_resolver.task_type, "ps")
179 simple_resolver = SimpleClusterResolver(base_cluster_spec, task_type="ps",
184 simple_resolver.task_type = "worker"
188 self.assertEqual(simple_resolver.task_type, "worker")
227 resolver1 = SimpleClusterResolver(cluster_spec_1, task_type="ps",
[all …]
Dgce_cluster_resolver_test.py179 task_type='testworker',
187 self.assertEqual(gce_cluster_resolver.task_type, 'testworker')
203 task_type='',
210 task_type='', task_id=0), 'grpc://10.1.2.3:8470')
217 task_type='custom',
273 task_type='worker',
282 task_type='worker',
291 task_type='ps',
333 task_type='testworker',
342 gce_cluster_resolver.task_type = 'foobar'
Dtfconfig_cluster_resolver_test.py159 self.assertEqual('ps', cluster_resolver.task_type)
178 cluster_resolver = TFConfigClusterResolver(task_type='ps', task_id=0)
181 self.assertEqual('ps', cluster_resolver.task_type)
184 cluster_resolver.task_type = 'worker'
189 self.assertEqual('worker', cluster_resolver.task_type)
208 self.assertEqual('123456', cluster_resolver.task_type)
303 self.assertEqual(resolver.num_accelerators(task_type='worker2', task_id=3),
/external/tensorflow/tensorflow/python/distribute/coordinator/
Dutils.py32 if not (cluster_resolver.task_type == 'worker' or
33 cluster_resolver.task_type == 'ps'):
35 cluster_resolver.task_type))
39 job_name=cluster_resolver.task_type,
44 cluster_resolver.task_type, cluster_resolver.task_id)
/external/tensorflow/tensorflow/python/keras/integration_test/
Dmulti_worker_tutorial_test.py36 def _is_chief(task_type, task_id): argument
44 return task_type == 'worker' and task_id == 0
54 def write_filepath(filepath, task_type, task_id): argument
57 if not _is_chief(task_type, task_id):
165 task_type, task_id = (strategy.cluster_resolver.task_type,
167 write_model_path = write_filepath(model_path, task_type, task_id)
170 if not _is_chief(task_type, task_id):
178 if tf.io.gfile.exists(write_model_path) != _is_chief(task_type, task_id):
186 write_checkpoint_dir = write_filepath(checkpoint_dir, task_type, task_id)
191 if not _is_chief(task_type, task_id):
[all …]
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/tpu/
Dtpu_cluster_resolver.py216 self.task_type = job_name
230 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
258 if task_type is not None and task_id is not None:
260 master = cluster_spec.task_address(task_type, task_id)
261 elif self.task_type is not None and self.task_id is not None:
263 master = cluster_spec.task_address(self.task_type, self.task_id)
266 job_tasks = cluster_spec.job_tasks(self.task_type)
278 return self.task_type
336 cluster_spec = {self.task_type: worker_list}
345 task_type=None, argument
/external/llvm-project/openmp/runtime/test/ompt/tasks/
Dtask_types_serialized.c12 int task_type; in print_task_type() local
14 ompt_get_task_info(0, &task_type, NULL, NULL, NULL, NULL); in print_task_type()
15 format_task_type(task_type, buffer); in print_task_type()
16 …intf("%" PRIu64 ": id=%d task_type=%s=%d\n", ompt_get_thread_data()->value, id, buffer, task_type); in print_task_type()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.cluster_resolver.-g-c-e-cluster-resolver.pbtxt19 name: "task_type"
24 …argspec: "args=[\'self\', \'project\', \'zone\', \'instance_group\', \'port\', \'task_type\', \'ta…
32 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'rpc_layer\'], varargs=None, keywords=None,…
36 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'config_proto\'], varargs=None, keywords=No…
Dtensorflow.distribute.cluster_resolver.-t-f-config-cluster-resolver.pbtxt19 name: "task_type"
24 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'rpc_layer\', \'environment\'], varargs=Non…
32 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'rpc_layer\'], varargs=None, keywords=None,…
36 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'config_proto\'], varargs=None, keywords=No…
Dtensorflow.distribute.cluster_resolver.-simple-cluster-resolver.pbtxt19 name: "task_type"
24 …argspec: "args=[\'self\', \'cluster_spec\', \'master\', \'task_type\', \'task_id\', \'environment\…
32 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'rpc_layer\'], varargs=None, keywords=None,…
36 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'config_proto\'], varargs=None, keywords=No…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.cluster_resolver.-g-c-e-cluster-resolver.pbtxt19 name: "task_type"
24 …argspec: "args=[\'self\', \'project\', \'zone\', \'instance_group\', \'port\', \'task_type\', \'ta…
32 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'rpc_layer\'], varargs=None, keywords=None,…
36 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'config_proto\'], varargs=None, keywords=No…
Dtensorflow.distribute.cluster_resolver.-t-f-config-cluster-resolver.pbtxt19 name: "task_type"
24 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'rpc_layer\', \'environment\'], varargs=Non…
32 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'rpc_layer\'], varargs=None, keywords=None,…
36 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'config_proto\'], varargs=None, keywords=No…
Dtensorflow.distribute.cluster_resolver.-simple-cluster-resolver.pbtxt19 name: "task_type"
24 …argspec: "args=[\'self\', \'cluster_spec\', \'master\', \'task_type\', \'task_id\', \'environment\…
32 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'rpc_layer\'], varargs=None, keywords=None,…
36 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'config_proto\'], varargs=None, keywords=No…
/external/llvm-project/lldb/test/API/python_api/type/
DTestTypeList.py93 task_type = target.FindFirstType('Task')
94 self.assertTrue(task_type)
95 self.DebugSBType(task_type)
98 task_ref_type = task_type.GetReferenceType()
104 task_pointer_type = task_type.GetPointerType()
122 self.assertTrue(task_type == task_head_pointee_type)

1234