Home
last modified time | relevance | path

Searched defs:worker_index (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/rust/android-crates-io/crates/rayon-core/src/sleep/
Dmod.rs37 worker_index: usize, field
125 let worker_index = idle_state.worker_index; in sleep() localVariable
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/coordinator/
H A Dcoordinator_context.py59 def worker_index(self): member in DispatchContext
H A Dcluster_coordinator.py740 def __init__(self, worker_index, device_name, cluster): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/data/service/
H A Dcommon.proto47 int64 worker_index = 12; field
H A Dauto_shard_rewriter.cc124 int64_t num_workers, int64_t worker_index) in AutoShardRewriter()
H A Dauto_shard_rewriter_test.cc73 const int64_t num_workers, const int64_t worker_index) { in GetTaskDef()
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/
H A Ddistribute.py366 num_replicas_per_worker, worker_index): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/service/
H A Dauto_shard_test.py96 def testRangeDataset_ShardHint(self, worker_index): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dinput_lib.py477 def compute_devices_for_worker(self, worker_index): argument
1183 def rebatch_fn(dataset, worker_index): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
H A Drebatch_dataset_test.py103 def get_batch_sizes_for_worker(worker_index): argument
/aosp_15_r20/external/libwebsockets/lib/misc/threadpool/
H A Dthreadpool.c77 int worker_index; member