Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/
Dtrt_convert_api.h42 size_t max_workspace_size_bytes = LLONG_MAX - 512; member
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py171 max_workspace_size_bytes=DEFAULT_TRT_MAX_WORKSPACE_SIZE_BYTES, argument
436 max_workspace_size_bytes=DEFAULT_TRT_MAX_WORKSPACE_SIZE_BYTES, argument
1149 max_workspace_size_bytes=DEFAULT_TRT_MAX_WORKSPACE_SIZE_BYTES, argument
1699 max_workspace_size_bytes=DEFAULT_TRT_MAX_WORKSPACE_SIZE_BYTES, argument
Dtrt_convert_test.py379 max_workspace_size_bytes=10 << 20, # Use a smaller workspace. argument
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dtrt_optimization_pass.h46 size_t max_workspace_size_bytes = 1 << 30; member
Dconvert_nodes.h130 int64 max_workspace_size_bytes; member
Dconvert_nodes.cc1227 size_t max_workspace_size_bytes, nvinfer1::IGpuAllocator* allocator, in BuildCudaEngine()
5831 int max_batch_size, size_t max_workspace_size_bytes, in ConvertGraphDefToEngine()