| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/integration_test/ |
| H A 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()
|
| H A 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()
|
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A 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()
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/autograd/ |
| H A 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()
|
| /aosp_15_r20/external/pytorch/torch/utils/data/_utils/ |
| H A D | worker.py | 178 def _generate_state(base_seed, worker_id): argument
241 worker_id, argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/autograd/context/ |
| H A D | container.cpp | 36 DistAutogradContainer& DistAutogradContainer::init(int64_t worker_id) { in init()
242 for (const auto& worker_id : workerIds) { in sendReleaseContextRpc() local
|
| /aosp_15_r20/prebuilts/build-tools/sysroots/arm-unknown-linux-musleabihf/include/linux/ |
| H A D | vhost_types.h | 31 unsigned int worker_id; member
35 unsigned int worker_id; member
|
| /aosp_15_r20/prebuilts/build-tools/sysroots/i686-unknown-linux-musl/include/linux/ |
| H A D | vhost_types.h | 31 unsigned int worker_id; member
35 unsigned int worker_id; member
|
| /aosp_15_r20/bionic/libc/kernel/uapi/linux/ |
| H A D | vhost_types.h | 31 unsigned int worker_id; member
35 unsigned int worker_id; member
|
| /aosp_15_r20/out/soong/ndk/sysroot/usr/include/linux/ |
| D | vhost_types.h | 31 unsigned int worker_id; member
35 unsigned int worker_id; member
|
| /aosp_15_r20/prebuilts/runtime/mainline/runtime/sdk/android/arm64/include/bionic/libc/kernel/uapi/linux/ |
| H A D | vhost_types.h | 31 unsigned int worker_id; member
35 unsigned int worker_id; member
|
| /aosp_15_r20/prebuilts/runtime/mainline/runtime/sdk/linux_bionic/include/bionic/libc/kernel/uapi/linux/ |
| H A D | vhost_types.h | 31 unsigned int worker_id; member
35 unsigned int worker_id; member
|
| /aosp_15_r20/external/kernel-headers/original/uapi/linux/ |
| H A D | vhost_types.h | 56 unsigned int worker_id; member
63 unsigned int worker_id; member
|
| /aosp_15_r20/prebuilts/runtime/mainline/runtime/sdk/android/x86/include/bionic/libc/kernel/uapi/linux/ |
| H A D | vhost_types.h | 31 unsigned int worker_id; member
35 unsigned int worker_id; member
|
| /aosp_15_r20/prebuilts/build-tools/sysroots/aarch64-unknown-linux-musl/include/linux/ |
| H A D | vhost_types.h | 31 unsigned int worker_id; member
35 unsigned int worker_id; member
|
| /aosp_15_r20/prebuilts/runtime/mainline/runtime/sdk/android/x86_64/include/bionic/libc/kernel/uapi/linux/ |
| H A D | vhost_types.h | 31 unsigned int worker_id; member
35 unsigned int worker_id; member
|
| /aosp_15_r20/prebuilts/runtime/mainline/runtime/sdk/android/arm/include/bionic/libc/kernel/uapi/linux/ |
| H A D | vhost_types.h | 31 unsigned int worker_id; member
35 unsigned int worker_id; member
|
| /aosp_15_r20/prebuilts/runtime/mainline/runtime/sdk/android/riscv64/include/bionic/libc/kernel/uapi/linux/ |
| H A D | vhost_types.h | 31 unsigned int worker_id; member
35 unsigned int worker_id; member
|
| /aosp_15_r20/prebuilts/build-tools/sysroots/x86_64-unknown-linux-musl/include/linux/ |
| H A D | vhost_types.h | 31 unsigned int worker_id; member
35 unsigned int worker_id; member
|
| /aosp_15_r20/external/pytorch/torch/utils/data/ |
| H A 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
|
| H A D | backward_compatibility.py | 10 def worker_init_fn(worker_id): argument
|
| /aosp_15_r20/external/crosvm/virtio_sys/src/ |
| H A D | vhost.rs | 92 pub worker_id: ::std::os::raw::c_uint, field
98 pub worker_id: ::std::os::raw::c_uint, field
|
| /aosp_15_r20/external/pytorch/test/ |
| H A 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
|