| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | shared_batch_scheduler_test.cc | 102 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 …]
|
| D | basic_batch_scheduler.h | 205 int max_enqueued_batches = 10; member
322 shared_scheduler_queue_options.max_enqueued_batches = in Create()
323 options.max_enqueued_batches; in Create()
|
| D | batch_resource_base.cc | 101 void RecordBatchParamMaxEnqueuedBatches(int64 max_enqueued_batches, in RecordBatchParamMaxEnqueuedBatches() argument
108 cell->GetCell(model_name, op_name)->Set(max_enqueued_batches); in RecordBatchParamMaxEnqueuedBatches()
187 batcher_queue_options_.max_enqueued_batches, GetModelName(context), in RegisterInput()
229 int32 max_enqueued_batches, const std::vector<int32>& allowed_batch_sizes, in GetBatcherQueueOptions() argument
233 batcher_queue_options.max_enqueued_batches = max_enqueued_batches; in GetBatcherQueueOptions()
260 int32 max_enqueued_batches, bool enable_large_batch_splitting, in GetAdaptiveBatcherQueueOptions() argument
265 batcher_queue_options.max_enqueued_batches = max_enqueued_batches; in GetAdaptiveBatcherQueueOptions()
|
| D | serial_device_batch_scheduler.h | 110 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()
|
| D | shared_batch_scheduler.h | 165 size_t max_enqueued_batches = 10; member
472 if (options.max_enqueued_batches < 0) { in AddQueue()
475 options.max_enqueued_batches); in AddQueue()
639 if (batches_.size() >= options_.max_enqueued_batches) { in ScheduleWithoutSplit()
700 options_.max_enqueued_batches - batches_.size(); in ScheduleWithSplit()
779 options_.max_enqueued_batches - batches_.size(); in SchedulingCapacity()
|
| D | batch_resource_base.h | 138 int32 max_enqueued_batches, const std::vector<int32>& allowed_batch_sizes,
143 int32 max_enqueued_batches, bool enable_large_batch_splitting,
|
| D | adaptive_shared_batch_scheduler.h | 142 int max_enqueued_batches = 10; member
413 if (options.max_enqueued_batches <= 0) { in AddQueue()
416 options.max_enqueued_batches); in AddQueue()
740 options_.max_enqueued_batches - num_enqueued_batches_; in SchedulingCapacityLocked()
|
| D | basic_batch_scheduler_test.cc | 72 options.max_enqueued_batches = 3; in TEST()
|
| D | basic_batch_scheduler_benchmark_test.cc | 345 scheduler_options.max_enqueued_batches = INT_MAX; // Unbounded queue. in RunThroughputBenchmark()
397 scheduler_options.max_enqueued_batches = INT_MAX; // Unbounded queue. in RunLatencyBenchmark()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | batch_ops.py | 36 max_enqueued_batches=10, argument
112 max_enqueued_batches=max_enqueued_batches,
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | BatchFunction.pbtxt | 32 name: "max_enqueued_batches"
116 name: "max_enqueued_batches"
|
| D | Batch.pbtxt | 98 name: "max_enqueued_batches"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | BatchFunction.pbtxt | 32 name: "max_enqueued_batches"
|
| D | Batch.pbtxt | 98 name: "max_enqueued_batches"
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 87 int32 batch_timeout_micros, int32 max_enqueued_batches, in Create() argument
101 batch_timeout_micros, max_enqueued_batches, in Create()
111 int32 max_enqueued_batches, const std::vector<int32>& allowed_batch_sizes, in Create() argument
121 max_batch_size, batch_timeout_micros, max_enqueued_batches, in Create()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BatchFunction.pbtxt | 43 name: "max_enqueued_batches"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.pbtxt | 837 …atch_size\', \'batch_timeout_micros\', \'allowed_batch_sizes\', \'max_enqueued_batches\', \'autogr…
|
| D | tensorflow.raw_ops.pbtxt | 317 …atch_size\', \'batch_timeout_micros\', \'grad_timeout_micros\', \'max_enqueued_batches\', \'allowe…
349 …eads\', \'max_batch_size\', \'batch_timeout_micros\', \'Tout\', \'max_enqueued_batches\', \'allowe…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops.td | 1361 DefaultValuedAttr<I64Attr, "10">:$max_enqueued_batches,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.pbtxt | 1737 …atch_size\', \'batch_timeout_micros\', \'allowed_batch_sizes\', \'max_enqueued_batches\', \'autogr…
|
| D | tensorflow.raw_ops.pbtxt | 317 …atch_size\', \'batch_timeout_micros\', \'grad_timeout_micros\', \'max_enqueued_batches\', \'allowe…
349 …eads\', \'max_batch_size\', \'batch_timeout_micros\', \'Tout\', \'max_enqueued_batches\', \'allowe…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 3318 name: "max_enqueued_batches"
3563 name: "max_enqueued_batches"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 8865 name: "max_enqueued_batches"
9126 name: "max_enqueued_batches"
|