HomeSort by relevance Sort by last modified time
    Searched refs:maximum_cached_engines (Results 1 - 9 of 9) sorted by null

  /external/tensorflow/tensorflow/contrib/tensorrt/python/
trt_convert.py 32 maximum_cached_engines=1,
46 maximum_cached_engines=maximum_cached_engines,
  /external/tensorflow/tensorflow/python/compiler/tensorrt/
trt_convert.py 469 maximum_cached_engines=1,
487 maximum_cached_engines: max number of cached TRT engines in dynamic TRT
493 should be <= maximum_cached_engines, and the dynamic TRT op will use
549 optimizer.parameter_map["maximum_cached_engines"].i = maximum_cached_engines
569 maximum_cached_engines=1,
598 maximum_cached_engines: max number of cached TRT engines in dynamic TRT
604 should be <= maximum_cached_engines, and the dynamic TRT op will use
681 if len(cached_engine_batches) > maximum_cached_engines:
683 "maximum_cached_engines items."
    [all...]
trt_convert_test.py 71 maximum_cached_engines=2,
85 "max_workspace_size_bytes", "precision_mode", "maximum_cached_engines",
97 self.assertEqual(2, trt_optimizer.parameter_map["maximum_cached_engines"].i)
176 maximum_cached_engines=1,
191 maximum_cached_engines=maximum_cached_engines,
383 maximum_cached_engines=2,
421 maximum_cached_engines=2, # This is noop, added just for testing.
  /external/tensorflow/tensorflow/python/compiler/tensorrt/test/
dynamic_input_shapes_test.py 85 maximum_cached_engines=10,
int32_test.py 65 maximum_cached_engines=1,
tf_trt_integration_test_base.py 61 "minimum_segment_size", "is_dynamic_op", "maximum_cached_engines",
183 maximum_cached_engines=1,
228 conversion_params.maximum_cached_engines,
310 maximum_cached_engines=conversion_params.maximum_cached_engines,
322 assert conversion_params.maximum_cached_engines == 1
biasadd_matmul_test.py 124 maximum_cached_engines=1,
  /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
convert_nodes.h 116 int maximum_cached_engines; member in class:tensorflow::tensorrt::convert::EngineInfo::EngineType
convert_graph.cc 165 parameters["maximum_cached_engines"].set_i(max_cached_engines);
565 .Attr("max_cached_engines_count", info.maximum_cached_engines)
    [all...]

Completed in 805 milliseconds