HomeSort by relevance Sort by last modified time
    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