Searched refs:num_consumers (Results 1 – 2 of 2) sorted by relevance
123 void LaunchConsumerThreads(int num_consumers, int cap_needed) { in LaunchConsumerThreads() argument124 barrier_.reset(new AlternatingBarrier(num_consumers)); in LaunchConsumerThreads()125 consumer_count_.resize(num_consumers, 0); in LaunchConsumerThreads()126 for (int i = 0; i < num_consumers; ++i) { in LaunchConsumerThreads()
578 int num_consumers = TF_OperationOutputNumConsumers(oper_out); in TF_OperationOutputConsumers_wrapper() local579 std::vector<TF_Input> consumers(num_consumers); in TF_OperationOutputConsumers_wrapper()580 TF_OperationOutputConsumers(oper_out, consumers.data(), num_consumers); in TF_OperationOutputConsumers_wrapper()582 std::vector<const char*> consumer_names(num_consumers); in TF_OperationOutputConsumers_wrapper()583 for (int i = 0; i < num_consumers; ++i) { in TF_OperationOutputConsumers_wrapper()