| /external/tensorflow/tensorflow/python/distribute/ |
| D | multi_worker_continuous_run_test.py | 74 def worker_step_fn(worker_id): argument
82 t_in = array_ops.ones(tensor_shape) * worker_id
95 worker_id = tf_config['task']['index']
97 worker_step_fn(worker_id)
107 def worker_step_fn(worker_id, num_dims): argument
116 initial_value = (array_ops.ones(tensor_shape) if worker_id == 0 else
128 worker_id = tf_config['task']['index']
130 worker_step_fn(worker_id, num_dims=(i + 1))
|
| D | input_lib_test.py | 1436 worker_id = multi_worker_util.id_in_cluster(cr.cluster_spec(), cr.task_type,
1439 if worker_id == 0:
1441 elif worker_id == 1:
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/quantiles/ |
| D | weighted_quantiles_stream_test.cc | 57 void GenerateFixedUniformSummary(int32 worker_id, int64 max_elements, in GenerateFixedUniformSummary() argument
67 void GenerateFixedNonUniformSummary(int32 worker_id, int64 max_elements, in GenerateFixedNonUniformSummary() argument
77 void GenerateRandUniformFixedWeightsSummary(int32 worker_id, int64 max_elements, in GenerateRandUniformFixedWeightsSummary() argument
81 random::PhiloxRandom philox(13 + worker_id); in GenerateRandUniformFixedWeightsSummary()
91 void GenerateRandUniformRandWeightsSummary(int32 worker_id, int64 max_elements, in GenerateRandUniformRandWeightsSummary() argument
95 random::PhiloxRandom philox(13 + worker_id); in GenerateRandUniformRandWeightsSummary()
133 void GenerateOneValue(int32 worker_id, int64 max_elements, double *total_weight, in GenerateOneValue() argument
140 void GenerateOneZeroWeightedValue(int32 worker_id, int64 max_elements, in GenerateOneZeroWeightedValue() argument
|
| /external/tensorflow/tensorflow/python/training/ |
| D | sync_replicas_optimizer_test.py | 39 for worker_id in range(num_workers):
41 is_chief = (worker_id == 0)
51 with ops.device("/job:worker/task:" + str(worker_id)):
52 grads_0 = constant_op.constant(0.1 + worker_id * 0.2)
53 grads_1 = constant_op.constant(0.9 + worker_id * 0.2)
57 [0.1 + worker_id * 0.2], shape=[1, 1]),
76 master=workers[worker_id].target,
|
| /external/rust/crates/tokio/src/runtime/thread_pool/ |
| D | idle.rs | 114 pub(super) fn unpark_worker_by_id(&self, worker_id: usize) { in unpark_worker_by_id()
118 if sleepers[index] == worker_id { in unpark_worker_by_id()
130 pub(super) fn is_parked(&self, worker_id: usize) -> bool { in is_parked()
132 sleepers.contains(&worker_id) in is_parked()
|
| /external/perfetto/test/stress_test/ |
| D | stress_producer.cc | 60 void WorkerMain(uint32_t worker_id);
111 void StressTestDataSource::Worker::WorkerMain(uint32_t worker_id) { in WorkerMain() argument
112 PERFETTO_DLOG("Worker %u starting", worker_id); in WorkerMain()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | bincount_op.cc | 66 [&](int64 start_ind, int64 limit_ind, int64 worker_id) { in Compute() argument
70 partial_bins(worker_id, value) = true; in Compute()
112 [&](int64 start_ind, int64 limit_ind, int64 worker_id) { in Compute() argument
117 partial_bins(worker_id, value) += weights(i); in Compute()
120 partial_bins(worker_id, value) += T(1); in Compute()
146 [&](int64 start_row, int64 end_row, int64 worker_id) { in Compute() argument
|
| /external/tensorflow/tensorflow/python/keras/integration_test/ |
| D | multi_worker_tutorial_test.py | 227 def extract_accuracy(worker_id, input_string): argument
229 r'\[worker\-{}\].*accuracy: (\d+\.\d+).*'.format(worker_id),
233 for worker_id in range(NUM_WORKERS):
235 lambda x: extract_accuracy(worker_id, x), # pylint: disable=cell-var-from-loop
|
| /external/perfetto/src/trace_processor/importers/ninja/ |
| D | ninja_log_parser.cc | 169 uint32_t worker_id = ++last_worker_id; in NotifyEndOfFile() local
173 auto utid = ctx_->process_tracker->UpdateThread(worker_id, job.build_id); in NotifyEndOfFile()
|
| /external/angle/src/tests/ |
| D | capture_replay_tests.py | 509 def SetWorkerId(self, worker_id): argument
510 self.trace_dir = "%s%d" % (TRACE_FOLDER, worker_id)
740 def RunTests(args, worker_id, job_queue, result_list, message_queue): argument
741 replay_build_dir = os.path.join(args.out_dir, 'Replay%d' % worker_id)
752 len(test_batch.tests), worker_id, job_queue.qsize()))
754 test_batch.SetWorkerId(worker_id)
893 for worker_id in range(worker_count):
895 target=RunTests, args=(args, worker_id, job_queue, result_list, message_queue))
|
| /external/python/cpython3/Lib/test/libregrtest/ |
| D | runtest_mp.py | 120 def __init__(self, worker_id, runner): argument
122 self.worker_id = worker_id
|
| /external/rust/crates/grpcio/src/ |
| D | cq.rs | 219 pub fn worker_id(&self) -> ThreadId { in worker_id() method
|
| /external/tensorflow/tensorflow/python/distribute/coordinator/ |
| D | cluster_coordinator.py | 312 def _select_worker_slice(worker_id, structured): argument
316 …return x._values[worker_id] if isinstance(x, PerWorkerValues) else x # pylint: disable=protected-…
|