tensorflow::
Summary
Public attributes |
|
|---|---|
allow_build_at_runtime = true
|
bool
|
convert_to_static_engine = true
|
bool
|
max_cached_engines = 1
|
int
|
max_workspace_size_bytes = 1 << 30
|
size_t
|
minimum_segment_size = 3
|
int
|
precision_mode = TrtPrecisionMode::FP32
|
TrtPrecisionMode
|
profile_strategy = ProfileStrategy::kRange
|
ProfileStrategy
|
use_calibration = true
|
bool
|
use_dynamic_shape = true
|
bool
|
Public attributes
allow_build_at_runtime
bool tensorflow::tensorrt::TfTrtConversionParams::allow_build_at_runtime = true
convert_to_static_engine
bool tensorflow::tensorrt::TfTrtConversionParams::convert_to_static_engine = true
max_cached_engines
int tensorflow::tensorrt::TfTrtConversionParams::max_cached_engines = 1
max_workspace_size_bytes
size_t tensorflow::tensorrt::TfTrtConversionParams::max_workspace_size_bytes = 1 << 30
minimum_segment_size
int tensorflow::tensorrt::TfTrtConversionParams::minimum_segment_size = 3
precision_mode
TrtPrecisionMode tensorflow::tensorrt::TfTrtConversionParams::precision_mode = TrtPrecisionMode::FP32
profile_strategy
ProfileStrategy tensorflow::tensorrt::TfTrtConversionParams::profile_strategy = ProfileStrategy::kRange
use_calibration
bool tensorflow::tensorrt::TfTrtConversionParams::use_calibration = true
use_dynamic_shape
bool tensorflow::tensorrt::TfTrtConversionParams::use_dynamic_shape = true