| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | union_find.cc | 85 ClusterBatchSize& ClusterBatchSize::SetMaxBatchSize(int max_batch_size) { in SetMaxBatchSize()
91 const absl::optional<int>& max_batch_size) { in SetMaxBatchSize()
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | batch_resource_base.cc | 92 void RecordBatchParamMaxBatchSize(int64 max_batch_size, in RecordBatchParamMaxBatchSize()
228 int32 num_batch_threads, int32 max_batch_size, int32 batch_timeout_micros, in GetBatcherQueueOptions()
241 std::vector<std::unique_ptr<BatchTask>>* output_tasks) -> Status { in GetBatcherQueueOptions()
259 int32 max_batch_size, int32 batch_timeout_micros, in GetAdaptiveBatcherQueueOptions()
277 std::vector<std::unique_ptr<BatchTask>>* output_tasks) -> Status { in GetAdaptiveBatcherQueueOptions()
381 int max_batch_size, std::vector<std::unique_ptr<BatchTask>>* output_tasks) { in SplitInputTask()
|
| D | basic_batch_scheduler.h | 183 int max_batch_size = 1000; member
|
| D | serial_device_batch_scheduler.h | 108 int max_batch_size = 1000; member
|
| D | adaptive_shared_batch_scheduler.h | 135 int max_batch_size = 1000; member
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | batch_ops.py | 33 max_batch_size, argument
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_convert.py | 234 max_batch_size=None, argument
318 max_batch_size=None, argument
377 max_batch_size=1, argument
1239 max_batch_size=1, argument
|
| D | trt_convert_test.py | 179 max_batch_size=1, argument
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| D | hexagon_delegate.h | 69 int max_batch_size; member
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.h | 37 size_t max_batch_size = 1; member
|
| D | convert_nodes.h | 113 absl::optional<int> max_batch_size; member
|
| D | convert_graph.cc | 429 int max_batch_size = info.max_batch_size.has_value() in CreateTRTNode() local
|
| D | convert_nodes.cc | 1450 TrtUniquePtrType<nvinfer1::ICudaEngine>* engine, int max_batch_size, in BuildCudaEngine()
6320 const GraphDef& gdef, TrtPrecisionMode precision_mode, int max_batch_size, in ConvertGraphDefToEngine()
|
| /external/mesa3d/src/gallium/winsys/i915/drm/ |
| D | i915_drm_winsys.h | 26 size_t max_batch_size; member
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | quantization_mnist_test.py | 114 def _GetGraphDef(self, use_trt, max_batch_size, model_dir): argument
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | batch_seq_config_test.cc | 105 void ApplyDelegate(int max_batch_size, in ApplyDelegate()
|
| /external/python/apitools/apitools/base/py/ |
| D | batch.py | 188 max_batch_size=None, batch_request_callback=None): argument
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 927 const auto max_batch_size = raw_static_engine->getMaxBatchSize(); in GetEngine() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 110 int32 max_batch_size, int32 batch_timeout_micros, in Create()
|
| /external/libtextclassifier/native/annotator/ |
| D | annotator.cc | 2983 const int max_batch_size = model_->selection_options()->batch_size(); in ModelClickContextScoreChunks() local
3092 const int max_batch_size = model_->selection_options()->batch_size(); in ModelBoundsSensitiveScoreChunks() local
|