Home
last modified time | relevance | path

Searched defs:cached_engine_batches (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensorrt/python/
Dtrt_convert.py33 cached_engine_batches=None, argument
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py470 cached_engine_batches=None, argument
570 cached_engine_batches=None, argument
790 cached_engine_batches=None, argument
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.h74 std::vector<int> cached_engine_batches; member
Dconvert_nodes.h117 std::vector<int> cached_engine_batches; member
Dconvert_graph.cc106 std::vector<int> cached_engine_batches, bool use_calibration) { in ConvertGraphDefToTensorRT()