Home
last modified time | relevance | path

Searched refs:num_workers_ (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/data/service/
Dtest_cluster.cc41 TestCluster::TestCluster(int num_workers) : num_workers_(num_workers) {} in TestCluster()
55 workers_.reserve(num_workers_); in Initialize()
56 worker_addresses_.reserve(num_workers_); in Initialize()
57 for (int i = 0; i < num_workers_; ++i) { in Initialize()
81 DCHECK_LT(index, num_workers_); in WorkerAddress()
Dtest_cluster.h44 int num_workers_; variable
/external/rust/crates/grpcio-sys/grpc/src/core/lib/iomgr/poller/
Deventmanager_libuv.h41 int num_workers() const { return num_workers_; } in num_workers()
42 void set_num_workers(int num) { num_workers_ = num; } in set_num_workers()
54 int num_workers_;
Deventmanager_libuv.cc25 grpc::experimental::LibuvEventManager::Options::Options() : num_workers_(-1) {} in Options()
27 : num_workers_(num_workers) {} in Options()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dauto_shard_dataset_op_test.cc34 num_workers_(num_workers), in AutoShardDatasetParams()
45 return CreateTensors<int64>(TensorShape({}), {{num_workers_}, {index_}}); in GetInputTensors()
72 int64 num_workers_; member in tensorflow::data::experimental::__anon8ac557bd0111::AutoShardDatasetParams
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dauto_shard.cc683 num_workers_ = config->parameter_map().at(kNumWorkersAttrName).i(); in Init()
696 if (num_workers_ < 1) { in Init()
698 " should be >= 1, currently ", num_workers_); in Init()
701 if (index_ < 0 || index_ >= num_workers_) { in Init()
703 num_workers_, ", currently ", index_); in Init()
718 TF_RETURN_IF_ERROR(OptimizeGraph(item, num_workers_, index_, in OptimizeAndCollectStats()
Dauto_shard.h50 int64 num_workers_;
/external/stressapptest/src/
Dworker.h82 WorkerStatus() : num_workers_(0), status_(RUN) {} in WorkerStatus()
88 num_workers_ += num_new_workers; in AddWorkers()
185 int num_workers_; variable
Dworker.cc134 num_workers_ + 1)); in Initialize()
217 sat_assert(0 == pthread_barrier_init(&pause_barrier_, NULL, num_workers_)); in RemoveSelf()
219 --num_workers_; in RemoveSelf()