Home
last modified time | relevance | path

Searched defs:max_workspace_size_bytes (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensorrt/python/
Dtrt_convert.py28 max_workspace_size_bytes=trt_convert.DEFAULT_TRT_MAX_WORKSPACE_SIZE_BYTES, argument
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py465 max_workspace_size_bytes=DEFAULT_TRT_MAX_WORKSPACE_SIZE_BYTES, argument
565 max_workspace_size_bytes=DEFAULT_TRT_MAX_WORKSPACE_SIZE_BYTES, argument
785 max_workspace_size_bytes=DEFAULT_TRT_MAX_WORKSPACE_SIZE_BYTES, argument
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.h62 size_t max_workspace_size_bytes = 1 << 30; member
Dconvert_nodes.h115 int64 max_workspace_size_bytes; member
Dconvert_graph.cc103 size_t max_batch_size, size_t max_workspace_size_bytes, in ConvertGraphDefToTensorRT()
Dconvert_nodes.cc4057 size_t max_workspace_size_bytes, in ConvertGraphDefToEngine()