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