tensorflow::tensorrt::TfTrtConversionParams

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