Home
last modified time | relevance | path

Searched refs:max_enqueued_batches (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/contrib/batching/python/ops/
Dbatch_ops.py62 max_enqueued_batches=10): argument
125 max_enqueued_batches=max_enqueued_batches,
143 max_enqueued_batches=10): argument
181 max_enqueued_batches=max_enqueued_batches,
/external/tensorflow/tensorflow/core/kernels/batching_util/
Dshared_batch_scheduler_test.cc102 queue_options.max_enqueued_batches = 2; in TEST()
160 queue_options.max_enqueued_batches = 2; in TEST()
222 queue_options.max_enqueued_batches = 2; in TEST()
278 queue_options.max_enqueued_batches = 2; in TEST()
324 queue_options.max_enqueued_batches = 2; in TEST()
376 queue_options.max_enqueued_batches = 100 /* give plenty of room */; in TEST()
411 for (const int max_enqueued_batches : {1, 2, 5}) { in TEST()
428 queue_options.max_enqueued_batches = max_enqueued_batches; in TEST()
433 EXPECT_EQ(max_enqueued_batches * 2, queue->SchedulingCapacity()); in TEST()
443 for (int i = 0; i < max_enqueued_batches; ++i) { in TEST()
[all …]
Dbasic_batch_scheduler.h177 int max_enqueued_batches = 10; member
232 shared_scheduler_queue_options.max_enqueued_batches = in Create()
233 options.max_enqueued_batches; in Create()
Dserial_device_batch_scheduler.h110 int max_enqueued_batches = 10; member
339 if (options.max_enqueued_batches <= 0) { in AddQueue()
342 options.max_enqueued_batches); in AddQueue()
499 if (num_enqueued_batches_ >= options_.max_enqueued_batches) { in Schedule()
541 options_.max_enqueued_batches - num_enqueued_batches_; in SchedulingCapacity()
Dshared_batch_scheduler.h160 size_t max_enqueued_batches = 10; member
406 if (options.max_enqueued_batches < 0) { in AddQueue()
409 options.max_enqueued_batches); in AddQueue()
542 if (batches_.size() >= options_.max_enqueued_batches) { in Schedule()
583 options_.max_enqueued_batches - batches_.size(); in SchedulingCapacity()
Dadaptive_shared_batch_scheduler.h127 int max_enqueued_batches = 10; member
371 if (options.max_enqueued_batches <= 0) { in AddQueue()
374 options.max_enqueued_batches); in AddQueue()
561 if (num_enqueued_batches_ >= options_.max_enqueued_batches) { in Schedule()
606 options_.max_enqueued_batches - num_enqueued_batches_; in SchedulingCapacity()
Dbasic_batch_scheduler_test.cc72 options.max_enqueued_batches = 3; in TEST()
Dbasic_batch_scheduler_benchmark_test.cc348 scheduler_options.max_enqueued_batches = INT_MAX; // Unbounded queue. in RunThroughputBenchmark()
398 scheduler_options.max_enqueued_batches = INT_MAX; // Unbounded queue. in RunLatencyBenchmark()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_BatchFunction.pbtxt43 name: "max_enqueued_batches"
/external/tensorflow/tensorflow/core/kernels/
Dbatch_kernels.cc211 int32 batch_timeout_micros, int32 max_enqueued_batches, in Create() argument
223 new_resource->batcher_queue_options_.max_enqueued_batches = in Create()
224 max_enqueued_batches; in Create()
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt2970 name: "max_enqueued_batches"
3208 name: "max_enqueued_batches"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt8865 name: "max_enqueued_batches"
9126 name: "max_enqueued_batches"
Dops_history.v1.pbtxt8865 name: "max_enqueued_batches"
9126 name: "max_enqueued_batches"