Home
last modified time | relevance | path

Searched defs:WorkerState (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dparallel_interleave_dataset_op.cc611 struct WorkerState { struct in tensorflow::data::experimental::ParallelInterleaveDatasetOp::Dataset::Iterator
613 std::vector<Tensor> input;
615 std::deque<OutputElem> outputs;
620 bool is_producing = false;
627 condition_variable cond_var;
629 inline bool MayHaveElements() const { in MayHaveElements()
634 void SetInputs(const Status& s, std::vector<Tensor> input_arguments) { in SetInputs()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dworker_session.cc111 struct WorkerState { struct in tensorflow::__anonde36c3c90111::WorkerFreeListCache
112 WorkerInterface* worker;
/external/pytorch/torch/distributed/elastic/agent/server/
Dapi.py190 class WorkerState(str, Enum): class
234 def is_running(state: "WorkerState") -> bool:
/external/crosvm/devices/src/virtio/vhost/user/device/
Dhandler.rs963 pub(crate) struct WorkerState<T, U> { struct
964 pub(crate) queue_task: TaskHandle<U>,
965 pub(crate) queue: T,
/external/pytorch/test/distributed/elastic/agent/server/test/
Dapi_test.py151 def monres(state: WorkerState):
/external/kotlinx.coroutines/kotlinx-coroutines-core/jvm/src/scheduling/
DCoroutineScheduler.kt999 enum class WorkerState { in <lambda>() class in kotlinx.coroutines.scheduling.CoroutineScheduler
/external/crosvm/devices/src/virtio/gpu/
Dmod.rs927 enum WorkerState { enum