/external/tensorflow/tensorflow/python/ops/ |
D | batch_ops.py | 33 max_batch_size, 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 | 141 size_t max_batch_size = 1000; member
|
D | serial_device_batch_scheduler.h | 108 int max_batch_size = 1000; member
|
D | adaptive_shared_batch_scheduler.h | 125 int max_batch_size = 1000; member
|
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
D | convert_graph.h | 39 size_t max_batch_size = 1; member
|
D | convert_graph.cc | 327 int max_batch_size, Graph* graph, in CreateTRTNode()
|
D | convert_nodes.cc | 1330 TrtUniquePtrType<nvinfer1::ICudaEngine>* engine, int max_batch_size, in BuildCudaEngine() 5681 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/compiler/xla/client/lib/ |
D | tridiagonal_test.cc | 49 const int64 max_batch_size = in XLA_TEST_P() local
|
/external/tensorflow/tensorflow/python/compiler/tensorrt/ |
D | trt_convert.py | 411 max_batch_size=1, argument 1159 max_batch_size=1, argument
|
D | trt_convert_test.py | 266 max_batch_size=1, argument
|
/external/python/apitools/apitools/base/py/ |
D | batch.py | 188 max_batch_size=None, batch_request_callback=None): argument
|
/external/tensorflow/tensorflow/compiler/tf2tensorrt/plugin/ |
D | plugin_cast.cu.cc | 103 PluginFormat float_format, int max_batch_size) override { in configurePlugin()
|
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
D | trt_engine_op.cc | 734 const auto max_batch_size = raw_static_engine->getMaxBatchSize(); in GetEngine() local
|
/external/libtextclassifier/native/annotator/ |
D | annotator.cc | 2722 const int max_batch_size = model_->selection_options()->batch_size(); in ModelClickContextScoreChunks() local 2833 const int max_batch_size = model_->selection_options()->batch_size(); in ModelBoundsSensitiveScoreChunks() local
|
/external/tensorflow/tensorflow/core/kernels/ |
D | batch_kernels.cc | 215 static Status Create(int32 num_batch_threads, int32 max_batch_size, in Create()
|