| /external/tensorflow/tensorflow/core/distributed_runtime/integration_test/ |
| D | c_api_multi_client_test.cc | 35 int worker_id, int cluster_size)> in StartWorkers()
50 auto worker_thread_fn = [&](int worker_id) { in StartWorkers()
97 int cluster_size) { in TEST()
125 int cluster_size) { in TEST()
161 int cluster_size) { in TEST()
|
| D | c_api_session_coordination_test.cc | 83 for (int worker_id = 0; worker_id < num_workers; worker_id++) { in TEST_P() local
173 for (int worker_id = 0; worker_id < num_workers; worker_id++) { in TEST_P() local
|
| D | c_api_coordination_test.cc | 121 auto worker_thread_fn = [&](int worker_id) { in TEST()
188 auto worker_thread_fn = [&](int worker_id) { in TEST()
278 auto worker_thread_fn = [&](int worker_id) { in TEST()
349 auto worker_thread_fn = [&](int worker_id) { in TEST()
|
| D | c_api_multi_client_function_test.cc | 207 auto worker_thread_fn = [&](int worker_id) { in TEST_P()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | multi_worker_continuous_run_test.py | 70 def worker_step_fn(worker_id): argument
103 def worker_step_fn(worker_id, num_dims): argument
|
| /external/pytorch/torch/utils/data/ |
| D | backward_compatibility.py | 10 def worker_init_fn(worker_id): argument
|
| D | dataloader.py | 104 def _sharding_worker_init_fn(worker_init_fn, world_size, rank_id, worker_id): argument
1494 def _mark_worker_as_unavailable(self, worker_id, shutdown=False): argument
|
| /external/pytorch/torch/csrc/distributed/autograd/ |
| D | init.cpp | 105 [](int64_t worker_id) { in dist_autograd_init()
128 [](int64_t worker_id) { DistAutogradContainer::init(worker_id); }, in dist_autograd_init()
|
| /external/pytorch/torch/utils/data/_utils/ |
| D | worker.py | 178 def _generate_state(base_seed, worker_id): argument
241 worker_id, argument
|
| /external/pytorch/torch/csrc/distributed/autograd/context/ |
| D | container.cpp | 36 DistAutogradContainer& DistAutogradContainer::init(int64_t worker_id) { in init()
242 for (const auto& worker_id : workerIds) { in sendReleaseContextRpc() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | bincount_op.cc | 68 [&](int64_t start_ind, int64_t limit_ind, int64_t worker_id) { in Compute()
141 [&](int64_t start_ind, int64_t limit_ind, int64_t worker_id) { in Compute()
181 [&](int64_t start_row, int64_t end_row, int64_t worker_id) { in Compute()
|
| /external/pytorch/test/distributed/elastic/timer/ |
| D | api_test.py | 50 def _reap_worker(self, worker_id): argument
|
| /external/pytorch/test/cpp/dist_autograd/ |
| D | test_dist_autograd.cpp | 42 rpc::worker_id_t worker_id = 1; in TEST_F() local
|
| /external/perfetto/src/trace_processor/importers/ninja/ |
| D | ninja_log_parser.cc | 163 uint32_t worker_id = static_cast<uint32_t>(workers.size()) + 1; in NotifyEndOfFile() local
|
| /external/kernel-headers/original/uapi/linux/ |
| D | vhost_types.h | 56 unsigned int worker_id; member
63 unsigned int worker_id; member
|
| /external/perfetto/test/stress_test/ |
| D | stress_producer.cc | 111 void StressTestDataSource::Worker::WorkerMain(uint32_t worker_id) { in WorkerMain()
|
| /external/crosvm/virtio_sys/src/ |
| D | vhost.rs | 94 pub worker_id: ::std::os::raw::c_uint, field
100 pub worker_id: ::std::os::raw::c_uint, field
|
| /external/rust/android-crates-io/crates/grpcio/src/ |
| D | cq.rs | 219 pub fn worker_id(&self) -> ThreadId { in worker_id() method
|
| /external/googleapis/google/dataflow/v1beta3/ |
| D | environment.proto | 233 string worker_id = 5; field
|
| /external/google-cloud-java/java-dataflow/proto-google-cloud-dataflow-v1beta3/src/main/proto/google/dataflow/v1beta3/ |
| D | environment.proto | 233 string worker_id = 5; field
|
| /external/pytorch/test/ |
| D | test_dataloader.py | 842 def disable_stderr(worker_id): argument
1028 def _test_worker_info_init_fn(worker_id): argument
1092 def init_fn(worker_id): argument
3389 def worker_init_fn(self, worker_id): argument
|
| /external/google-cloud-java/java-tpu/proto-google-cloud-tpu-v1/src/main/proto/google/cloud/tpu/v1/ |
| D | cloud_tpu.proto | 619 string worker_id = 4; field
|
| /external/googleapis/google/cloud/tpu/v1/ |
| D | cloud_tpu.proto | 619 string worker_id = 4; field
|
| /external/googleapis/google/cloud/tpu/v2/ |
| D | cloud_tpu.proto | 782 string worker_id = 4; field
|
| /external/google-cloud-java/java-tpu/proto-google-cloud-tpu-v2/src/main/proto/google/cloud/tpu/v2/ |
| D | cloud_tpu.proto | 775 string worker_id = 4; field
|