| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | union_find.cc | 85 ClusterBatchSize& ClusterBatchSize::SetMaxBatchSize(int max_batch_size) { in SetMaxBatchSize()
91 const std::optional<int>& max_batch_size) { in SetMaxBatchSize()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | batch_ops.py | 30 max_batch_size, argument
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | batch_resource_base.cc | 172 void RecordBatchParamMaxBatchSize(int64_t max_batch_size, in RecordBatchParamMaxBatchSize()
331 int32_t num_batch_threads, int32_t max_batch_size, in GetBatcherQueueOptions()
345 std::vector<std::unique_ptr<BatchTask>>* output_tasks) -> Status { in GetBatcherQueueOptions()
363 int32_t max_batch_size, int32_t batch_timeout_micros, in GetAdaptiveBatcherQueueOptions()
381 std::vector<std::unique_ptr<BatchTask>>* output_tasks) -> Status { in GetAdaptiveBatcherQueueOptions()
491 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 | shared_batch_scheduler_test.cc | 156 std::vector<std::unique_ptr<FakeTask>>* output_tasks) -> Status { in get_split_func()
825 std::vector<std::unique_ptr<FakeTask>>* output_tasks) -> Status { in CreateQueues()
|
| D | adaptive_shared_batch_scheduler.h | 146 int max_batch_size = 1000; member
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_convert.py | 279 max_batch_size=None, argument
376 max_batch_size=None, argument
435 max_batch_size=1, argument
1698 max_batch_size=1, argument
|
| D | trt_convert_test.py | 181 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 | trt_optimization_pass.h | 45 size_t max_batch_size = -1; member
|
| D | convert_graph.cc | 461 int max_batch_size = info.max_batch_size.has_value() in CreateTRTNode() local
693 const EngineInfo& info, int max_batch_size, in CreateStaticEngine()
|
| D | convert_nodes.h | 131 std::optional<int> max_batch_size; member
|
| D | convert_nodes.cc | 1226 TrtUniquePtrType<nvinfer1::ICudaEngine>* engine, int max_batch_size, in BuildCudaEngine()
5831 int max_batch_size, size_t max_workspace_size_bytes, 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 | 148 def _GetGraphDef(self, use_trt, max_batch_size, model_dir): argument
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | runtime_fallback_batch_tf_opkernels.cc | 115 static Status Create(int32_t num_batch_threads, int32_t max_batch_size, in Create()
140 int32_t max_batch_size, int32_t batch_timeout_micros, in Create()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | batch_seq_config_test.cc | 106 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 | 1186 int max_batch_size = 1; in GetEngine() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 159 int32_t max_batch_size, int32_t batch_timeout_micros, in Create()
|
| /external/libtextclassifier/native/annotator/ |
| D | annotator.cc | 3000 const int max_batch_size = model_->selection_options()->batch_size(); in ModelClickContextScoreChunks() local
3109 const int max_batch_size = model_->selection_options()->batch_size(); in ModelBoundsSensitiveScoreChunks() local
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 5693 size_t max_batch_size = 2; in TEST() local
|