OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:max_batch_size
(Results
1 - 8
of
8
) sorted by null
/external/mesa3d/src/gallium/winsys/i915/drm/
i915_drm_winsys.h
26
size_t
max_batch_size
;
member in struct:i915_drm_winsys
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
convert_graph.h
61
size_t
max_batch_size
= 1;
member in struct:tensorflow::tensorrt::convert::ConversionParams
79
// -
max_batch_size
: maximum batch size which can be used for inference for
85
size_t
max_batch_size
, size_t max_workspace_size_bytes,
/external/tensorflow/tensorflow/core/kernels/batching_util/
basic_batch_scheduler.h
149
int
max_batch_size
= 1000;
member in struct:tensorflow::serving::BasicBatchScheduler::Options
154
// even if the batch's size is below '
max_batch_size
'.
229
shared_scheduler_queue_options.
max_batch_size
= options.
max_batch_size
;
shared_batch_scheduler.h
139
size_t
max_batch_size
= 1000;
member in struct:tensorflow::serving::SharedBatchScheduler::QueueOptions
144
// even if the batch's size is below '
max_batch_size
'.
253
size_t max_task_size() const { return options_.
max_batch_size
; }
397
if (options.
max_batch_size
== 0) {
398
return errors::InvalidArgument("
max_batch_size
must be positive; was ",
399
options.
max_batch_size
);
529
if ((*task)->size() > options_.
max_batch_size
) {
532
options_.
max_batch_size
);
541
if (batches_.back()->size() + (*task)->size() > options_.
max_batch_size
) {
585
options_.
max_batch_size
- batches_.back()->size()
[
all
...]
adaptive_shared_batch_scheduler.h
125
int
max_batch_size
= 1000;
member in struct:tensorflow::serving::AdaptiveSharedBatchScheduler::QueueOptions
257
size_t max_task_size() const override { return options_.
max_batch_size
; }
367
if (options.
max_batch_size
<= 0) {
368
return errors::InvalidArgument("
max_batch_size
must be positive; was ",
369
options.
max_batch_size
);
550
if (size > options_.
max_batch_size
) {
553
options_.
max_batch_size
);
560
current_batch_->size() + size > options_.
max_batch_size
) {
604
current_batch_ ? options_.
max_batch_size
- current_batch_->size() : 0;
607
return spare_batches * options_.
max_batch_size
+ current_batch_capacity
[
all
...]
serial_device_batch_scheduler.h
108
int
max_batch_size
= 1000;
member in struct:tensorflow::serving::SerialDeviceBatchScheduler::QueueOptions
226
size_t max_task_size() const override { return options_.
max_batch_size
; }
335
if (options.
max_batch_size
<= 0) {
336
return errors::InvalidArgument("
max_batch_size
must be positive; was ",
337
options.
max_batch_size
);
489
if (size > options_.
max_batch_size
) {
492
options_.
max_batch_size
);
498
current_batch_->size() + size > options_.
max_batch_size
) {
539
current_batch_ ? options_.
max_batch_size
- current_batch_->size() : 0;
542
return spare_batches * options_.
max_batch_size
+ current_batch_capacity
[
all
...]
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
trt_engine_op.cc
564
const auto
max_batch_size
= raw_static_engine->getMaxBatchSize();
local
565
// Static engine will have
max_batch_size
for batch size so that all inputs
570
engine_input_shapes[i].set_dim(0,
max_batch_size
);
585
if (
max_batch_size
< batch_size) {
/external/libtextclassifier/annotator/
annotator.cc
2164
const int
max_batch_size
= model_->selection_options()->batch_size();
local
2275
const int
max_batch_size
= model_->selection_options()->batch_size();
local
[
all
...]
Completed in 374 milliseconds