Searched refs:num_consumers (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_allocator_retry_test.cc123 void LaunchConsumerThreads(int num_consumers, int cap_needed) { argument
124 barrier_.reset(new AlternatingBarrier(num_consumers));
125 consumer_count_.resize(num_consumers, 0);
126 for (int i = 0; i < num_consumers; ++i) {
/external/tensorflow/tensorflow/python/client/
H A Dtf_session_helper.cc405 int num_consumers = TF_OperationOutputNumConsumers(oper_out); local
406 std::vector<TF_Input> consumers(num_consumers);
407 TF_OperationOutputConsumers(oper_out, consumers.data(), num_consumers);
409 std::vector<const char*> consumer_names(num_consumers);
410 for (int i = 0; i < num_consumers; ++i) {

Completed in 112 milliseconds