Home
last modified time | relevance | path

Searched defs:max_batch_size (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dbatch_ops.py33 max_batch_size, argument
/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbasic_batch_scheduler.h149 int max_batch_size = 1000; member
Dshared_batch_scheduler.h141 size_t max_batch_size = 1000; member
Dserial_device_batch_scheduler.h108 int max_batch_size = 1000; member
Dadaptive_shared_batch_scheduler.h125 int max_batch_size = 1000; member
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.h39 size_t max_batch_size = 1; member
Dconvert_graph.cc327 int max_batch_size, Graph* graph, in CreateTRTNode()
Dconvert_nodes.cc1330 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/
Di915_drm_winsys.h26 size_t max_batch_size; member
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dquantization_mnist_test.py114 def _GetGraphDef(self, use_trt, max_batch_size, model_dir): argument
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dtridiagonal_test.cc49 const int64 max_batch_size = in XLA_TEST_P() local
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py411 max_batch_size=1, argument
1159 max_batch_size=1, argument
Dtrt_convert_test.py266 max_batch_size=1, argument
/external/python/apitools/apitools/base/py/
Dbatch.py188 max_batch_size=None, batch_request_callback=None): argument
/external/tensorflow/tensorflow/compiler/tf2tensorrt/plugin/
Dplugin_cast.cu.cc103 PluginFormat float_format, int max_batch_size) override { in configurePlugin()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc734 const auto max_batch_size = raw_static_engine->getMaxBatchSize(); in GetEngine() local
/external/libtextclassifier/native/annotator/
Dannotator.cc2722 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/
Dbatch_kernels.cc215 static Status Create(int32 num_batch_threads, int32 max_batch_size, in Create()