| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | basic_batch_scheduler_benchmark_test.cc | 342 int num_batch_threads) { in RunThroughputBenchmark() argument
347 scheduler_options.num_batch_threads = num_batch_threads; in RunThroughputBenchmark()
353 static void ThroughputBM_ZeroTimeout(int iters, int num_batch_threads) { in ThroughputBM_ZeroTimeout() argument
354 RunThroughputBenchmark(iters, 0 /* 0 ms timeout */, num_batch_threads); in ThroughputBM_ZeroTimeout()
365 static void ThroughputBM_SmallTimeout(int iters, int num_batch_threads) { in ThroughputBM_SmallTimeout() argument
366 RunThroughputBenchmark(iters, 1 * 1000 /* 1 ms timeout */, num_batch_threads); in ThroughputBM_SmallTimeout()
377 static void ThroughputBM_LargeTimeout(int iters, int num_batch_threads) { in ThroughputBM_LargeTimeout() argument
379 num_batch_threads); in ThroughputBM_LargeTimeout()
397 scheduler_options.num_batch_threads = kNumBatchThreads; in RunLatencyBenchmark()
|
| D | adaptive_shared_batch_scheduler.h | 94 int64 num_batch_threads = port::NumSchedulableCPUs();
308 if (options.num_batch_threads < 1) { in Create()
310 options.num_batch_threads); in Create()
317 if (options.min_in_flight_batches_limit > options.num_batch_threads) { in Create()
320 ") must be <= num_batch_threads (", options.num_batch_threads, ")"); in Create()
327 if (options.initial_in_flight_batches_limit > options.num_batch_threads) { in Create()
332 options.num_batch_threads, ")"); in Create()
360 GetEnv(), options.thread_pool_name, options.num_batch_threads)); in AdaptiveSharedBatchScheduler()
447 options_.num_batch_threads) { in MaybeScheduleClosedBatch()
509 static_cast<double>(options_.num_batch_threads)); in CallbackWrapper()
|
| D | shared_batch_scheduler_test.cc | 69 for (int num_batch_threads : {1, 2, 3}) { in TEST()
93 options.num_batch_threads = num_batch_threads; in TEST()
154 options.num_batch_threads = 2; in TEST()
215 options.num_batch_threads = 1; in TEST()
272 options.num_batch_threads = 2; in TEST()
315 options.num_batch_threads = 2; in TEST()
369 options.num_batch_threads = 1; in TEST()
422 options.num_batch_threads = 1; in TEST()
493 options.num_batch_threads = 2; in TEST()
548 options.num_batch_threads = 1; in TEST()
|
| D | serial_device_batch_scheduler.h | 76 int64 num_batch_threads = port::NumSchedulableCPUs();
265 if (options.num_batch_threads < 1) { in Create()
267 options.num_batch_threads); in Create()
274 if (options.initial_in_flight_batches_limit > options.num_batch_threads) { in Create()
279 options.num_batch_threads, ")"); in Create()
313 env(), options.thread_pool_name, options.num_batch_threads)); in SerialDeviceBatchScheduler()
437 std::min(in_flight_batches_limit_, options_.num_batch_threads); in ProcessBatches()
|
| D | basic_batch_scheduler.h | 170 int num_batch_threads = port::NumSchedulableCPUs();
221 shared_scheduler_options.num_batch_threads = options.num_batch_threads; in Create()
|
| D | adaptive_shared_batch_scheduler_test.cc | 71 options.num_batch_threads = 0; in TEST()
77 options.num_batch_threads = 5; in TEST()
88 options.num_batch_threads = 3; in TEST()
93 options.num_batch_threads = 3; in TEST()
|
| D | serial_device_batch_scheduler_test.cc | 73 options.num_batch_threads = 0; in TEST()
79 options.num_batch_threads = 5; in TEST()
94 options.num_batch_threads = 3; in TEST()
137 options.num_batch_threads = 3; in TEST()
|
| D | shared_batch_scheduler.h | 112 int num_batch_threads = port::NumSchedulableCPUs();
364 if (options.num_batch_threads < 1) { in Create()
366 options.num_batch_threads); in Create()
441 for (int i = 0; i < options.num_batch_threads; ++i) { in SharedBatchScheduler()
|
| D | basic_batch_scheduler_test.cc | 71 options.num_batch_threads = 1; in TEST()
|
| /external/tensorflow/tensorflow/contrib/batching/python/ops/ |
| D | batch_ops.py | 58 def batch_function(num_batch_threads, argument
121 num_batch_threads=num_batch_threads,
137 def batch_function_v1(num_batch_threads, argument
178 num_batch_threads=num_batch_threads,
|
| D | batch_ops_test.py | 49 [inp], num_batch_threads=1, max_batch_size=2,
89 [inp], num_batch_threads=1, max_batch_size=10,
121 num_batch_threads=1,
159 [inp0, inp1], num_batch_threads=1, max_batch_size=2,
172 [inp], num_batch_threads=1, max_batch_size=10,
274 num_batch_threads=1,
304 num_batch_threads=1,
337 num_batch_threads=1,
376 [inp], num_batch_threads=1, max_batch_size=2,
418 [inp], num_batch_threads=1, max_batch_size=2,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BatchFunction.pbtxt | 23 name: "num_batch_threads"
107 num_batch_threads=1,
|
| D | api_def_Batch.pbtxt | 23 num_batch_threads: Number of scheduling threads for processing batches of work.
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 210 static Status Create(int32 num_batch_threads, int32 max_batch_size, in Create() argument
218 batcher_options.num_batch_threads = num_batch_threads; in Create()
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 36594 func Batch(scope *Scope, in_tensors []tf.Output, num_batch_threads int64, max_batch_size int64, bat…
36598 …attrs := map[string]interface{}{"num_batch_threads": num_batch_threads, "max_batch_size": max_batc…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 2962 name: "num_batch_threads"
3196 name: "num_batch_threads"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 8787 name: "num_batch_threads"
8857 name: "num_batch_threads"
9114 name: "num_batch_threads"
|
| D | ops_history.v1.pbtxt | 8787 name: "num_batch_threads"
8857 name: "num_batch_threads"
9114 name: "num_batch_threads"
|