Home
last modified time | relevance | path

Searched refs:default_max_batch_size (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dtf_trt_integration_test_base.py681 default_max_batch_size, argument
788 self.assertTrue(default_max_batch_size is None or
789 engine_max_batch_size == default_max_batch_size or
860 default_max_batch_size=self.GetMaxBatchSize(run_params))
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc333 int default_max_batch_size, Graph* graph, in CreateTRTNode() argument
431 : default_max_batch_size; in CreateTRTNode()