Home
last modified time | relevance | path

Searched defs:task_id (Results 1 – 25 of 60) sorted by relevance

123

/external/tensorflow/tensorflow/python/distribute/cluster_resolver/
Dcluster_resolver.py115 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
136 task_id=None, argument
242 def task_id(self): member in ClusterResolver
287 def task_id(self, task_id): member in ClusterResolver
324 def __init__(self, cluster_spec, master='', task_type=None, task_id=None, argument
349 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
377 def task_id(self): member in SimpleClusterResolver
385 def task_id(self, task_id): argument
394 task_id=None, argument
573 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
[all …]
Dtfconfig_cluster_resolver.py86 task_id=None, argument
114 def task_id(self): member in TFConfigClusterResolver
126 def task_id(self, task_id): member in TFConfigClusterResolver
146 task_id=None, argument
164 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
Dgce_cluster_resolver.py74 task_id=0, argument
174 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
192 def task_id(self): member in GCEClusterResolver
202 def task_id(self, task_id): argument
Dsagemaker_cluster_resolver.py96 task_id=None, argument
124 def task_id(self): member in SageMakerClusterResolver
136 def task_id(self, task_id): argument
154 def num_accelerators(self, task_type=None, task_id=None, config_proto=None): argument
171 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
Dslurm_cluster_resolver.py373 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
397 task_id=None, argument
/external/libbrillo/brillo/message_loops/
Dmessage_loop_test.cc84 TaskId task_id = in TYPED_TEST() local
94 TaskId task_id = in TYPED_TEST() local
105 TaskId task_id = in TYPED_TEST() local
127 TaskId task_id; in TYPED_TEST() local
131 [](bool* cancel_result, MessageLoop* loop, TaskId* task_id) { in TYPED_TEST()
Dbase_message_loop.cc84 TaskId task_id = NextTaskId(); in PostDelayedTask() local
100 bool BaseMessageLoop::CancelTask(TaskId task_id) { in CancelTask()
172 void BaseMessageLoop::OnRanPostedTask(MessageLoop::TaskId task_id) { in OnRanPostedTask()
Dmessage_loop_utils.cc16 MessageLoop::TaskId task_id = loop->PostDelayedTask( in MessageLoopRunUntil() local
Dbase_message_loop.h81 MessageLoop::TaskId task_id; member
/external/tensorflow/tensorflow/python/distribute/
Dmulti_worker_util.py58 task_id): argument
112 def is_chief(cluster_spec=None, task_type=None, task_id=None): argument
152 def collective_leader(cluster_spec, task_type, task_id): argument
207 def id_in_cluster(cluster_spec, task_type, task_id): argument
Dmulti_process_runner.py251 def _continuously_readline_from_sub(self, pipe_r, task_type, task_id): argument
266 task_id, argument
405 task_id, argument
463 def get_process_id(self, task_type, task_id): argument
469 def get_process_exit_code(self, task_type, task_id): argument
487 def process_exists(self, task_type, task_id): argument
666 def terminate(self, task_type, task_id): argument
753 def _message_checking_func(self, task_type, task_id): argument
987 def _pool_runner_worker(task_type, task_id, initializer, conn): argument
1016 def _run_contained(task_type, task_id, fn, args, kwargs): argument
[all …]
Dparameter_server_strategy_test.py75 task_id=None, argument
115 def _get_test_objects(self, task_type, task_id, num_gpus): argument
123 def _test_device_assignment_distributed(self, task_type, task_id, num_gpus): argument
223 self, task_type, task_id, num_gpus): argument
372 def _test_simple_increment(self, task_type, task_id, num_gpus): argument
436 def _test_minimize_loss_graph(self, task_type, task_id, num_gpus): argument
524 task_id, argument
Dcollective_all_reduce_strategy_test.py71 task_id=None, argument
105 def _get_test_object(self, task_type, task_id, num_gpus=0): argument
113 def _test_minimize_loss_graph(self, task_type, task_id, num_gpus): argument
185 def _test_variable_initialization(self, task_type, task_id, num_gpus): argument
215 task_id, argument
348 def _worker_fn(task_type, task_id, required_gpus): argument
Ddistribute_coordinator.py116 task_id, argument
291 def task_id(self): member in _WorkerContext
330 task_id, argument
387 task_id=None, argument
530 strategy, eval_strategy, session_config, cluster_spec, task_type, task_id): argument
638 task_id=None, argument
Dmulti_worker_test_base.py311 def kill_task(self, task_type, task_id): argument
326 def start_task(self, task_type, task_id): argument
533 def _run_client(self, client_fn, task_type, task_id, num_gpus, eager_mode, argument
694 def _run_task_in_thread(self, task_fn, cluster_spec, task_type, task_id, argument
756 def _run_task_in_process(self, cmd_args, cluster_spec, task_type, task_id): argument
/external/libchrome/base/task/
Dcancelable_task_tracker_unittest.cc112 CancelableTaskTracker::TaskId task_id = task_tracker_.PostTask( in TEST_F() local
130 CancelableTaskTracker::TaskId task_id = in TEST_F() local
149 CancelableTaskTracker::TaskId task_id = in TEST_F() local
168 CancelableTaskTracker::TaskId task_id = task_tracker_.PostTaskAndReply( in TEST_F() local
189 CancelableTaskTracker::TaskId task_id = in TEST_F() local
360 void TryCancel(CancelableTaskTracker::TaskId task_id, in TryCancel()
372 CancelableTaskTracker::TaskId task_id = in TEST_F() local
391 CancelableTaskTracker::TaskId task_id = in TEST_F() local
/external/tensorflow/tensorflow/python/keras/integration_test/
Dmulti_worker_tutorial_test.py36 def _is_chief(task_type, task_id): argument
47 def _get_temp_dir(dirpath, task_id): argument
54 def write_filepath(filepath, task_type, task_id): argument
/external/tensorflow/tensorflow/core/data/service/
Dcommon.proto20 int64 task_id = 4; field
36 int64 task_id = 2; field
Djournal.proto86 int64 task_id = 1; field
95 int64 task_id = 1; field
102 int64 task_id = 1; field
Ddispatcher_state.cc152 int64 task_id = create_pending_task.task_id(); in CreatePendingTask() local
187 int64 task_id = create_task.task_id(); in CreateTask() local
201 int64 task_id = finish_task.task_id(); in FinishTask() local
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtpu_client.cc93 std::vector<std::unique_ptr<PjRtStreamExecutorDevice>> devices, int task_id) in PjRtTpuClient()
158 int task_id = topology.IdForHost(core.host_coordinates()); in GetTpuDevices() local
217 int task_id = platform->GetTpuHostLocation().Id(); in GetTpuClient() local
Dtpu_client.h33 int task_id, const std::array<int, 3>& coords, in PjRtTpuDevice() argument
/external/tensorflow/tensorflow/lite/delegates/gpu/common/memory_management/
Dgreedy_by_breadth_assignment.cc37 TaskId task_id; member
57 for (size_t task_id = 0; task_id < task_profiles.size(); ++task_id) { in GreedyByBreadthAssignment() local
/external/libchrome/base/
Ddeferred_sequenced_task_runner_unittest.cc28 int task_id) in ExecuteTaskOnDestructor()
39 void ExecuteTask(int task_id) { in ExecuteTask()
44 void PostExecuteTask(int task_id) { in PostExecuteTask()
/external/autotest/client/site_tests/platform_PrinterPpds/
Dtest_multithreaded_processor.py33 def _thread(self, task_id): argument

123