Home
last modified time | relevance | path

Searched refs:queue_options (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dshared_batch_scheduler_test.cc99 SharedBatchScheduler<FakeTask>::QueueOptions queue_options; in TEST() local
100 queue_options.max_batch_size = 10; in TEST()
101 queue_options.batch_timeout_micros = 10 * 1000 * 1000; // 10 seconds in TEST()
102 queue_options.max_enqueued_batches = 2; in TEST()
105 scheduler->AddQueue(queue_options, queue_0_callback, &queue_0)); in TEST()
108 scheduler->AddQueue(queue_options, queue_1_callback, &queue_1)); in TEST()
157 SharedBatchScheduler<FakeTask>::QueueOptions queue_options; in TEST() local
158 queue_options.max_batch_size = 10; in TEST()
159 queue_options.batch_timeout_micros = 10 * 1000 * 1000; // 10 seconds in TEST()
160 queue_options.max_enqueued_batches = 2; in TEST()
[all …]
Dserial_device_batch_scheduler_test.cc234 SerialDeviceBatchScheduler<FakeTask>::QueueOptions queue_options; in TEST() local
238 queue_options.max_batch_size = 1000; in TEST()
239 TF_ASSERT_OK(scheduler->AddQueue(queue_options, queue_callback, &queue1)); in TEST()
240 queue_options.max_batch_size = 1000; in TEST()
241 TF_ASSERT_OK(scheduler->AddQueue(queue_options, queue_callback, &queue2)); in TEST()
242 queue_options.max_batch_size = 100; in TEST()
243 TF_ASSERT_OK(scheduler->AddQueue(queue_options, queue_callback, &queue3)); in TEST()
Dadaptive_shared_batch_scheduler_test.cc235 AdaptiveSharedBatchScheduler<FakeTask>::QueueOptions queue_options; in TEST() local
238 queue_options.max_batch_size = 1000; in TEST()
239 TF_ASSERT_OK(scheduler->AddQueue(queue_options, queue_callback, &queue1)); in TEST()
240 queue_options.max_batch_size = 100; in TEST()
241 TF_ASSERT_OK(scheduler->AddQueue(queue_options, queue_callback, &queue2)); in TEST()
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine.cc169 RunOptions queue_options = run_options_; in Run() local
170 if (queue_options.trace_level() >= RunOptions::HARDWARE_TRACE) { in Run()
171 queue_options.set_trace_level(RunOptions::SOFTWARE_TRACE); in Run()
179 session_.get(), queue_options)); in Run()