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/ |
D | mod.rs | 37 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 D | coordinator_context.py | 59 def worker_index(self): member in DispatchContext
|
H A D | cluster_coordinator.py | 740 def __init__(self, worker_index, device_name, cluster): argument
|
/aosp_15_r20/external/tensorflow/tensorflow/core/data/service/ |
H A D | common.proto | 47 int64 worker_index = 12; field
|
H A D | auto_shard_rewriter.cc | 124 int64_t num_workers, int64_t worker_index) in AutoShardRewriter()
|
H A D | auto_shard_rewriter_test.cc | 73 const int64_t num_workers, const int64_t worker_index) { in GetTaskDef()
|
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/ |
H A D | distribute.py | 366 num_replicas_per_worker, worker_index): argument
|
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/service/ |
H A D | auto_shard_test.py | 96 def testRangeDataset_ShardHint(self, worker_index): argument
|
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
H A D | input_lib.py | 477 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 D | rebatch_dataset_test.py | 103 def get_batch_sizes_for_worker(worker_index): argument
|
/aosp_15_r20/external/libwebsockets/lib/misc/threadpool/ |
H A D | threadpool.c | 77 int worker_index; member
|