| /external/tensorflow/tensorflow/contrib/tensorrt/custom_plugin_examples/ |
| D | inc_op_plugin.h | 59 int max_batch_size) override { in configure()
69 size_t getWorkspaceSize(int max_batch_size) const override { return 0; } in getWorkspaceSize()
|
| /external/tensorflow/tensorflow/contrib/batching/python/ops/ |
| D | batch_ops.py | 59 max_batch_size, argument
138 max_batch_size, argument
|
| /external/tensorflow/tensorflow/contrib/tensorrt/python/ |
| D | trt_convert.py | 27 max_batch_size=1, argument
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_convert.py | 464 max_batch_size=1, argument
564 max_batch_size=1, argument
784 max_batch_size=1, argument
|
| D | trt_convert_test.py | 173 max_batch_size=1, argument
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | basic_batch_scheduler.h | 149 int max_batch_size = 1000; member
|
| D | shared_batch_scheduler.h | 139 size_t max_batch_size = 1000; member
|
| D | adaptive_shared_batch_scheduler.h | 125 int max_batch_size = 1000; member
|
| D | serial_device_batch_scheduler.h | 108 int max_batch_size = 1000; member
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.h | 61 size_t max_batch_size = 1; member
|
| D | convert_graph.cc | 103 size_t max_batch_size, size_t max_workspace_size_bytes, in ConvertGraphDefToTensorRT()
402 int max_batch_size, Graph* graph, in CreateTRTNode()
|
| D | convert_nodes.cc | 4056 const GraphDef& gdef, TrtPrecisionMode precision_mode, int max_batch_size, in ConvertGraphDefToEngine()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/plugin/ |
| D | trt_plugin.cc | 52 int max_batch_size) { in configure()
|
| /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 | 113 def _GetGraphDef(self, use_trt, max_batch_size, model_dir): argument
|
| /external/python/apitools/apitools/base/py/ |
| D | batch.py | 185 max_batch_size=None, batch_request_callback=None): argument
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 564 const auto max_batch_size = raw_static_engine->getMaxBatchSize(); in GetEngine() local
|
| /external/libtextclassifier/annotator/ |
| D | annotator.cc | 2164 const int max_batch_size = model_->selection_options()->batch_size(); in ModelClickContextScoreChunks() local
2275 const int max_batch_size = model_->selection_options()->batch_size(); in ModelBoundsSensitiveScoreChunks() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 210 static Status Create(int32 num_batch_threads, int32 max_batch_size, in Create()
|