Searched defs:max_batch_size (Results 1 - 7 of 7) sorted by relevance

/external/mesa3d/src/gallium/winsys/i915/drm/
H A Di915_drm_winsys.h26 size_t max_batch_size; member in struct:i915_drm_winsys
/external/tensorflow/tensorflow/core/kernels/batching_util/
H A Dbasic_batch_scheduler.h149 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;
H A Dadaptive_shared_batch_scheduler.h151 int max_batch_size = 1000; member in struct:tensorflow::serving::AdaptiveSharedBatchScheduler::QueueOptions
302 size_t max_task_size() const override { return options_.max_batch_size; }
424 if (options.max_batch_size <= 0) {
425 return errors::InvalidArgument("max_batch_size must be positive; was ",
426 options.max_batch_size);
600 if (size > options_.max_batch_size) {
603 options_.max_batch_size);
609 current_batch_->size() + size > options_.max_batch_size) {
650 current_batch_ ? options_.max_batch_size - current_batch_->size() : 0;
653 return spare_batches * options_.max_batch_size
[all...]
H A Dshared_batch_scheduler.h138 int max_batch_size = 1000; member in struct:tensorflow::serving::SharedBatchScheduler::QueueOptions
143 // even if the batch's size is below 'max_batch_size'.
252 size_t max_task_size() const { return options_.max_batch_size; }
396 if (options.max_batch_size <= 0) {
397 return errors::InvalidArgument("max_batch_size must be positive; was ",
398 options.max_batch_size);
528 if ((*task)->size() > options_.max_batch_size) {
531 options_.max_batch_size);
540 if (batches_.back()->size() + (*task)->size() > options_.max_batch_size) {
584 options_.max_batch_size
[all...]
/external/tensorflow/tensorflow/contrib/tensorrt/convert/
H A Dconvert_graph.cc116 size_t max_batch_size, // Max batch size that engine will be created for
154 max_batch_size, max_workspace_size_bytes, graph_properties,
200 const std::vector<string>& output_names, size_t max_batch_size,
261 output_names, subgraph_node_ids, max_batch_size,
113 ConvertSubGraphToTensorRT( const std::vector<string>& output_names, const std::set<int>& subgraph_node_ids, size_t max_batch_size, size_t max_workspace_size_bytes, const tensorflow::grappler::GraphProperties& graph_properties, tensorflow::Graph* graph) argument
198 ConvertGraphDefToTensorRT( const tensorflow::GraphDef& graph_def, const std::vector<string>& output_names, size_t max_batch_size, size_t max_workspace_size_bytes, tensorflow::GraphDef* new_graph_def) argument
/external/libtextclassifier/
H A Dtext-classifier.cc1386 const int max_batch_size = model_->selection_options()->batch_size(); local
1391 batch_start < span_of_interest.second; batch_start += max_batch_size) {
1393 std::min(batch_start + max_batch_size, span_of_interest.second);
1397 all_features.reserve(max_batch_size * cached_features.OutputFeaturesSize());
1497 const int max_batch_size = model_->selection_options()->batch_size(); local
1502 batch_start += max_batch_size) {
1503 const int batch_end = std::min(batch_start + max_batch_size,
1508 all_features.reserve(max_batch_size * cached_features.OutputFeaturesSize());
/external/tensorflow/tensorflow/core/kernels/
H A Dbatch_kernels.cc209 static Status Create(int32 num_batch_threads, int32 max_batch_size, argument
220 new_resource->batcher_queue_options_.max_batch_size = max_batch_size;
515 OP_REQUIRES_OK(c, c->GetAttr("max_batch_size", &max_batch_size_));
563 "final entry in allowed_batch_sizes must equal max_batch_size");

Completed in 306 milliseconds