Searched defs:max_batch_size (Results 1 - 7 of 7) sorted by relevance
/external/mesa3d/src/gallium/winsys/i915/drm/ |
H A D | i915_drm_winsys.h | 26 size_t max_batch_size; member in struct:i915_drm_winsys
|
/external/tensorflow/tensorflow/core/kernels/batching_util/ |
H A D | 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;
|
H A D | adaptive_shared_batch_scheduler.h | 151 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 D | shared_batch_scheduler.h | 138 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 D | convert_graph.cc | 116 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 D | text-classifier.cc | 1386 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 D | batch_kernels.cc | 209 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