Searched refs:num_consumers_ (Results 1 – 3 of 3) sorted by relevance
126 : num_consumers_(num_consumers), in RoundRobinTaskRunner()127 buffer_(num_consumers_), in RoundRobinTaskRunner()128 prefetch_thread_(std::move(iterator), num_consumers_) { in RoundRobinTaskRunner()139 if (req.consumer_index() >= num_consumers_) { in ValidateRequest()142 ", but the task is configured for only ", num_consumers_, " consumers"); in ValidateRequest()184 if (current_round_ < req.round_index() && round.size() == num_consumers_) { in PrepareRound()194 num_consumers_) { in PrepareRound()
149 const int64 num_consumers_; variable
105 num_consumers_(num_consumers), in Dataset()189 b->AddScalar(num_consumers_.value_or(-1), &num_consumers)); in AsGraphDefInternal()276 dataset()->num_consumers_, job_client_id_); in Initialize()780 bool StrictRoundRobin() { return dataset()->num_consumers_.has_value(); } in StrictRoundRobin()845 const absl::optional<int64> num_consumers_; member in tensorflow::data::DataServiceDatasetOp::Dataset