Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/
H A 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
H A Dtrt_convert_test.py379 max_workspace_size_bytes=10 << 20, # Use a smaller workspace. argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/
H A Dtrt_convert_api.h42 size_t max_workspace_size_bytes = LLONG_MAX - 512; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dtrt_optimization_pass.h46 size_t max_workspace_size_bytes = 1 << 30; member
H A Dconvert_nodes.h130 int64 max_workspace_size_bytes; member
H A 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()