Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/
H A Dtrt_convert_api.h37 struct TfTrtConversionParams { struct
42 size_t max_workspace_size_bytes = LLONG_MAX - 512;
48 TrtPrecisionMode precision_mode = TrtPrecisionMode::FP32;
54 int minimum_segment_size = 3;
58 int max_cached_engines = 1;
68 bool use_calibration = true;
74 bool use_dynamic_shape = true;
79 ProfileStrategy profile_strategy = ProfileStrategy::kRange;
86 bool allow_build_at_runtime = true;
116 const TfTrtConversionParams& conv_params); argument