Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert_windows.py83 max_workspace_size_bytes=DEFAULT_TRT_MAX_WORKSPACE_SIZE_BYTES, argument
Dtrt_convert.py156 max_workspace_size_bytes=DEFAULT_TRT_MAX_WORKSPACE_SIZE_BYTES, argument
378 max_workspace_size_bytes=DEFAULT_TRT_MAX_WORKSPACE_SIZE_BYTES, argument
1240 max_workspace_size_bytes=DEFAULT_TRT_MAX_WORKSPACE_SIZE_BYTES, argument
Dtrt_convert_test.py381 max_workspace_size_bytes=10 << 20, # Use a smaller workspace. argument
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.h38 size_t max_workspace_size_bytes = 1 << 30; member
Dconvert_nodes.h112 int64 max_workspace_size_bytes; member
Dconvert_nodes.cc1451 size_t max_workspace_size_bytes, nvinfer1::IGpuAllocator* allocator, in BuildCudaEngine()
6321 size_t max_workspace_size_bytes, in ConvertGraphDefToEngine()