HomeSort by relevance Sort by last modified time
    Searched defs:num_batch_threads (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/batching_util/
basic_batch_scheduler.h 60 // - the maximum time to process ceil(max_enqueued_batches/num_batch_threads)
170 int num_batch_threads = port::NumSchedulableCPUs(); member in struct:tensorflow::serving::BasicBatchScheduler::Options
221 shared_scheduler_options.num_batch_threads = options.num_batch_threads;
shared_batch_scheduler.h 111 int num_batch_threads = port::NumSchedulableCPUs(); member in struct:tensorflow::serving::SharedBatchScheduler::Options
363 if (options.num_batch_threads < 1) {
364 return errors::InvalidArgument("num_batch_threads must be positive; was ",
365 options.num_batch_threads);
440 for (int i = 0; i < options.num_batch_threads; ++i) {
adaptive_shared_batch_scheduler.h 106 int64 num_batch_threads = port::NumSchedulableCPUs(); member in struct:tensorflow::serving::AdaptiveSharedBatchScheduler::Options
347 if (options.num_batch_threads < 1) {
348 return errors::InvalidArgument("num_batch_threads must be positive; was ",
349 options.num_batch_threads);
377 if (options.initial_in_flight_batches_limit > options.num_batch_threads) {
381 ") should not be larger than num_batch_threads (",
382 options.num_batch_threads, ")");
413 GetEnv(), options.thread_pool_name, options.num_batch_threads));
516 static_cast<double>(options_.num_batch_threads));

Completed in 263 milliseconds